train script added