Added smoothing feature to VAE