Added stuff for hw6.