flake8ed test_tree, and tree.py