Docs updated based on PR comments