Corrections code to pull request