Working on VAE demo
more vae