FGM notebook update: using subset of training data