Update tree logic... root node not properly update