add generator, more data for more realism