Small changes to the PR