diff --git a/Shangai-B_dataset.png b/Shangai-B_dataset.png
deleted file mode 100644
index 69f237e..0000000
Binary files a/Shangai-B_dataset.png and /dev/null differ
diff --git a/visualize.py b/visualize.py
index 2afd6ec..18abce4 100644
--- a/visualize.py
+++ b/visualize.py
@@ -4,7 +4,7 @@ import matplotlib.pyplot as plt
 
 cur_dir = os.getcwd()
 #relative_path = 'out-pack/1784598-stn-newloss-shb/test_stats.h5'
-relative_paths = ['out-pack/filter_test_base-sha.h5', 'out-pack/filter_test_oldloss-sha.h5', 'out-pack/filter_test_newloss-sha.h5', 'out-pack/filter_test_base-shb.h5','out-pack/filter_test_newloss-shb.h5','out-pack/filter_test_oldloss-shb.h5']
+relative_paths = ['out-pack/filter_test_base-sha.h5', 'out-pack/filter_test_oldloss-sha.h5', 'out-pack/filter_test_newloss-sha.h5', 'out-pack/filter_test_base-shb.h5','out-pack/filter_test_oldloss-shb.h5','out-pack/filter_test_newloss-shb.h5']
 labels = ['base', 'oldloss', 'newloss']
 first_group = relative_paths[:3]
 second_group =relative_paths[3:]
@@ -64,7 +64,7 @@ plt.suptitle("Shangai-B dataset")
 plt.tight_layout()
 
 # Save the graph
-plt.savefig('Shangai-B_dataset.png')
+plt.savefig('Shangai-B_datasets.png')
 
 # Show plot
 plt.show()