update on the data generation code