Updating to new DecisionTree constructor