Added some stuff for lab 6.