Improvements to docs