Added generator for test data