Update VisualizationRunner.java