updates and starting proper testing on semi-supervised VAE