Small improvements to docs