Changed scripts for new TrainSetup