Touched up the code.