Update visualization.js