Adding headlines for train and test data

In the stats file