Update StressVisualizer.java