(oops of course the commands above need a `plt.show()`)