Hotfix for slight notebook issues raised by @jstac.

Fixes minor issues with `examples/solving_initial_value_problems.ipynb`.