hw4 - almost done