Understanding disentangling in $\beta$-VAE