Added script and plot for problem 5