Touched up code.