Update visualize.js