Needs `plt.show()` at the end