Adding train2 to replicate winning solution.