Add decision tree creation API