clean up to pass pep8