Add plt.grid()