shrink function

Subset only required columns