HW2 part 3 complete