add container for train data