Added getTrainController