HW4 part2 Update