Cleanup of the code.