implemented ELBO computation