HW2 Problem #1 is done