revised HW4 submission