update datagen script