Moved the UIButtons to the general TrainViewController from the different lines