Apply code review feedback