HW 4 done (I think)