updated the data generation code part
Revert "updated the data generation code part"

This reverts commit 8d680c96c1e417f7edf950df73d06632804a3205.