added test script to train multiple models