Some improvements to the docs