HW2 part 2 Completed