Update trees.py