Added decision tree diagram