Refactor decision tree class

[omarshammas]

The decision tree class uses a dictionary under hood which makes the code harder to read and as a result maintain. Refactor by adding a node class. makes it clearer.