Before I revamp tree_test.py