added variational distribution for latent space