Add ITrainBase and Train data structure.