Improve some of the docs