renamed train_mod2 to train_mod