Diagnosing and Enhancing VAE Models