wip: fix token processing for leaf nodes, visualize comm graph in shawn