clean up the code.