Completing HW2 Part 3