Add test data generator