- change the generator api to accept more data input variants