small changes for hw6