made changes to data generator