fixed bug in decision tree