add refactored version of decision tree