HW4 part2, first draft