Improved plots in toy_dynamics_tis.ipynb.