This is just so everything looks nice in nbviewer. I added plt.tight_layout() to the interactive_plot context manager, which makes plots fill the output cell better.