Small pep8 changes