Small pep8 clean up