Small pep8 cleanup