Update the visualization