pep8 fixes

pep8 fixes