Fitting the decision tree