minor cleanup w/ pep8