updated train.sh for pltrain