Fix pep8 issues

Fix pep8 issues