pep8 cleanup

pep8 cleanup