Added some tests on the train.