Touch up the code.