code cleanup (pep8)