HW4 edit