Updates for HW4p2