Update the DecisionTree class.