pick_cols function

Helper function to pick columns from a data frame