Update Visualization.java