Pep8 fix up