Added missing tree stuff