updated some of the visualization code