add data generation script
add script for data generation