working on decision tree