notebook version of plot_solutions.py