Added HW3 Prob 2 stuff