pep8 clean-up