HW4 done (close enough!)