pep8 cleanup 

Minor pep8 cleanup.