HW4 part 2 first update