small thing: maybe 'plt.plot' -> 'plt.plot()' (then it looks the same as the automatic names)