GMM notebook

and a few minor changes/fixes to make it work