Rewrote decision tree API and modified tests, all tests pass.
Cleaning up some tests.
Modified example to use new API.