work on generative modeling