Small pep8 corrections