Update tree.swift