add HW2 prob1