Wrote a plot_solution function.