Updated the visualization scripts.