adding python file for data generation