Update test data generation