fixed train/test sciprts