Clean up code for pep8 compliance.