code cleanup
pep8 cleanup