add decision tree part