Add column selector to `replace` op; refactor