fixed HW0 p1 and p2