Added graph plotting for new dataset