Added code to generate data and graph it