hw4, pr2, done i think