pep 8 refactoring