Update tree.go