pep 8 refactor