Moved to TrainViewController