removed test_mod and train_mod