Minor pep8 fixes