HW4 rough draft