Trying to figure out building the decision tree...