add HW2 prob2