Update TrainContainer.java