Implement code-review suggestions