few minor corrections from the PR review