Added column in data set to identify test/train data.