Update Hw2 part2