Add hw2 part #2