updating visualization code
updating visualization code