Small pep8 fixes