pep8 clean up