Update TreeModel.php