diff --git a/canny_shock_finder.py b/canny_shock_finder.py
index 2f97c41..e9e316f 100644
--- a/canny_shock_finder.py
+++ b/canny_shock_finder.py
@@ -478,7 +478,7 @@ def canny_shock_finder(time_list, pressure_list, sigma = 4, derivative_threshold
         except Exception as e:
             print (e)
             print ("There was an issue plotting the result.")
-            mplt.close('all')
+            #mplt.close('all')
 
     if not return_processing_lists:
         # just return the found arrival time/times
diff --git a/main.py b/main.py
index 1542cf1..ab0c5ae 100644
--- a/main.py
+++ b/main.py
@@ -166,6 +166,7 @@ x2_out = canny_shock_finder(time, (gData["x2"][4][:] - gData["x2"][4][0]) * 0.01
 
 print(x2_out)
 print("Done")
+pltKeyClose()
 
 #print("Graphing Data")
 #genGraph(data[loaded_data[0]])