Finalized the visualizer for now