changes to II.2.2 (now with training data)