Refactor Decision Tree API
Merge