diff --git a/image_ref/main.py b/image_ref/main.py
index 9ac992811634d342d5047a61d2cfa3ba9e6bdac7..3dda6c88a54560e971cfdb686b24b463907f8474 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -133,12 +133,12 @@ def run_duo(args):
 
 
     plt.show()
-    plt.savefig('output/training_plot_contrastive_noise_{}_lr_{}_model_{}.png'.format(args.noise_threshold,args.lr,args.model))
+    plt.savefig('output/training_plot_contrastive_{}.png'.format(args.prop))
 
     #load and evaluate best model
     load_model(model, args.save_path)
-    make_prediction_duo(model,data_test_batch, 'output/confusion_matrix_contractive_noise_{}_lr_{}_model_{}.png'.format(args.noise_threshold,args.lr,args.model),
-                        'output/confidence_matrix_contractive_noise_{}_lr_{}_model_{}.png'.format(args.noise_threshold,args.lr,args.model))
+    make_prediction_duo(model,data_test_batch, 'output/confusion_matrix_contractive_{}.png'.format(args.prop),
+                        'output/confidence_matrix_contractive_{}.png'.format(args.prop))
 
 
 def make_prediction_duo(model, data, f_name, f_name2):