edited the visualization code
edited the visualization code