diff --git a/data_visualisation.py b/data_visualisation.py
index 57e17c2..2f7dd17 100644
--- a/data_visualisation.py
+++ b/data_visualisation.py
@@ -44,6 +44,8 @@ blue.append((48/255, 181/255, 197/255))
 black = []
 black.append((25/255, 24/255, 21/255))
 
+plt.yscale('log')
+plt.grid(True)
 
 x = df_transistors.year
 y = df_transistors.transistors
@@ -65,14 +67,12 @@ x = df_cores.year
 y = df_cores.cores
 plt.scatter(x, y, marker = "o", color = black)
 
-plt.yscale('log')
-
-# plt.legend(loc="upper right", ["Cold", "Medium", "Hot"])
 plt.legend(["Transistors (thousands)", 
             "Single-thread SPECint (thousands)", 
             "Frequency (MHz)", 
             "Typical power (Watts)", 
             "Number of logical cores"], 
-           loc = "upper left")
+           loc = "upper left",
+           fontsize = "small")
 
 plt.show()
\ No newline at end of file