minor updates to GMM notebook