Draft implementation of decision tree