Correcting pep8 issues