Add support for decision tree

[smohadjer]

I don't think this requires any changes in frontend, but if you think otherwise let me know.