Touched up some code.