updates to the tree code