Completed HW1 part 2