Update TreeController.php