updated HW4 pt3