Redoing problem 2 plot