Fixed bug in derivation of lambda. More work on ELBO