Add header to displayTrain