Cleanup the code.