started writing/tracking unit tests, checked ELBO