forgot small change to tree diagram