Fix EpiTrain description