Finished abstract and conclusion.

python_scripts/normalization.py

@@ -21,9 +21,17 @@ def plot():
         "data/probability_deviation_no_slits.txt",
         "data/probability_deviation_slits.txt",
     ]
-    for file in files:
+    labels = [
+        "none",
+        "double-slit"
+    ]
+    colors = sns.color_palette("mako", n_colors=2)
+
+    fig, ax = plt.subplots() 
+
+    for file, label, color in zip(files, labels, colors):
         with open(file) as f:
-            lines = f.readlines();
+            lines = f.readlines()
             x = []
             arr = []
             for line in lines:
@@ -31,10 +39,13 @@ def plot():
                 x.append(float(tmp[0]))
                 arr.append(float(tmp[1]))
 
-            plt.plot(x,arr)
+            ax.plot(x, arr, label=label, color=color)
+        
+        ax.legend(title="Barrier")
+        ax.set_xlabel("Time (t)")
+        ax.set_ylabel("Deviation")
+    fig.savefig("latex/images/probability_deviation.pdf", bbox_inches="tight")
 
-        plt.savefig("latex/images/probability_deviation.pdf")
 
 if __name__ == "__main__":
-    plot()
-
+    plot()