cleanup for pep8