Adding in output node labeling for the tree