Added decision tree - basic structure for future use