HW 5 finish part 1