Update tree_controller.rb