Finished implementing building the decision tree