cleaning up for pep8