pep8 refactoring