Minor pep8 cleanup