Completing HW2 Part 1