Add changes to tree base class.