Script to generate training data