HW4 part1 finished