Added decision tree support