HW4 After mods