diff --git a/src/timing.py b/src/timing.py
index 3f6a10f..63b7457 100644
--- a/src/timing.py
+++ b/src/timing.py
@@ -2,6 +2,7 @@ import matplotlib.pyplot as plt
 import numpy as np
 
 def main():
+    fig, ax = plt.subplots()
     x = []
     gen_alg = []
     spec_alg = []
@@ -13,11 +14,11 @@ def main():
             gen_alg.append(float(gen_i))
             spec_alg.append(float(spec_i))
 
-        plt.plot(x, gen_alg, label=f"general algorithm")
-        plt.plot(x, spec_alg, label=f"general algorithm")
+        ax.plot(x, gen_alg, label=f"General algorithm")
+        ax.plot(x, spec_alg, label=f"Special algorithm")
 
-    plt.legend()
-    plt.savefig("../latex/images/problem10.pdf")
+    ax.legend()
+    fig.savefig("../latex/images/problem10.pdf")
 
 if __name__ == "__main__":
     main()