attempt to add tree diagram