hw4 fixed bugs