Clean up my code.