improve visualizer