question about train/test columns