Minor fixes to Train.java