diff --git a/simulation_TREPR.m b/simulation_TREPR.m
index af29e93..963bf4b 100644
--- a/simulation_TREPR.m
+++ b/simulation_TREPR.m
@@ -25,6 +25,7 @@ eventdata.manExp.Harmonic = 0; % zeroth harmonic
 eventdata.manExp.Temperature = params.Triplett_pop;
 eventdata.simSys = params.Manual_Sys; %initialize simulation parameters
 eventdata.simExp = eventdata.manExp;
+eventdata.SaveStat = 0; %counter if save button was pushed
 
 %% creating UI
 fig = figure;
@@ -349,10 +350,16 @@ uiwait(fig)
         eventdata.simExp.Temperature = eventdata.BestSys.Temperature;
         %save fitted parameters to struct
         params.Fitted_Simulation = eventdata.BestSys;
+        eventdata.SaveStat = eventdata.SaveStat + 1;
     end
     
     function SimDoneButtonPushed(~,~)
-        close 'Simulation'
+        if eventdata.SaveStat == 0
+            params.Fitted_Simulation = eventdata.BestSys;
+            close 'Simulation'
+        else
+            close 'Simulation'
+        end
     end
  
 end
\ No newline at end of file