Update visualize.rb