Added script/plot for problem 2