WIP making the visualization okay.