WIP: test_tree.py now passes \o/