Use a map to make the viz code more semantic

The only downside to this is that it introduces a .each over a 1-element
Map, but that's not actually bad.