HW 4 update