train script