pep8 code cleanup