Implement code review feedback