this is just for the data generator right?