HW2 part 1 update and correction