HW2 part 2 completed