Update FlowVisualizer.java