changes for HW4