Updated the visualization