update visualize.js