finish the decision tree part