Further work on visualizer
Almost finalized visualizer