TrainController modification (OK)