diff --git a/compare_graph_maker.py b/compare_graph_maker.py
index 3b4d855..5ff1b9b 100644
--- a/compare_graph_maker.py
+++ b/compare_graph_maker.py
@@ -2,9 +2,9 @@ import os
 import matplotlib.pyplot as plt
 
 datasets = {
-    "cleaned": "games_march2025_cleaned",
-    "cleaned_2k": "games_march2025_cleaned_2k",
-    "cleaned_10k": "games_march2025_cleaned_10k"
+    #"cleaned": "games_march2025_cleaned",
+    "cleaned_2k": "games_march2025_cleaned_2k_i3k",
+    "cleaned_10k": "games_march2025_cleaned_10k_i3k"
 }
 # def results
 results = {}
@@ -28,14 +28,14 @@ x = range(len(models))
 
 plt.figure(figsize=(12,6))
 #plt.bar([i - 0.25 for i in x], [results["cleaned"][m] for m in models], width=0.25, label="cleaned")
-plt.bar(x, [results["cleaned_2k"][m] for m in models], width=0.5)#, label="cleaned_2k")
-#plt.bar([i + 0.25 for i in x], [results["cleaned_10k"][m] for m in models], width=0.25, label="cleaned_10k")
+plt.bar(x, [results["cleaned_2k"][m] for m in models], width=0.25, label="2k Dataset")
+plt.bar([i + 0.25 for i in x], [results["cleaned_10k"].get(m,0) for m in models], width=0.25, label="10k Dataset")
 
 plt.xticks(x, models, rotation=90)
 plt.ylim(0, 1) # min max
 plt.ylabel("Weighted F1-Score")
 plt.title("Model Performance across Datasets")
-#plt.legend()
+plt.legend()
 plt.tight_layout()
 plt.savefig('compare_graph_latest.png')
 plt.show()