Task 2 Data Generation Part