Add generateData function