Move visualizer and drawing to it's own class