Refactor to only process first column