add code to generate data