cleaned visualizer class