HW2 part 2 complete