Update for new Visualizer class.