Update train.sh

Continuing getting rid of code for POS features