Updated test data gen script