Add HW 2 question 1