Part2 1st attempt to implement Decision Tree