HW4 p2 update