Update solve_and_plot.sh