Clean Up the code. Remove comments