Re-arrange some things (split-off TRAIN.sh script, change options)