add data generation script (messy)