Another pep8 clean up