Add TRAIN_NUMBER for some function