Implementing feedback from review