First draft of the visualizer