Update tree.lua