Update visualization.R