Updating documentation based on comments from pull request