Add tree_test.py