Minor improvements to docs