change decision tree to multi class classification