Improvements to the docs