Made some pep8 corrections