Small fix ups to docs and comments