Finding disentangled representations using VAE