[chore] Refactored column-selector

Updated to standards and added section heading
Fixed tests