Add column description

Also fixes the column selection from the original data set