add file header into the train model