Pep8 corrections

Some minor adjustments