Add hw 1 problem 2