Visualizer-related changes

Checking in the new visualizer