more testing on the train class