Replace everything from `autoscale_view` to here with `plt.margins(0.1)`.