Add script for test data generation