HW2_Prob_2 done