Added some stuff for hw6.