Refactor some of the docs