Refactor the docs