Update TreeGrid.php