Polished visualizer a little bit