Adding the visualization piece