Touching up some code.