refactor
pep8