Implemented unit tests for decision trees