Got the visualization to work