Clean up the code.