Adding initial data_generator script