Update visualizer.rb