HW4 changes