This is the training set for decision tree