Finish HW1 part 4