updated the data viz part