Making changes following pull request feedback