Addressing code review feedback