small changes HW4