changes to train parallel