A little bit more stable generative model

Still not promising