Minor pep8 changes