added decision tree stuff