finish hw4 cleanup