Add docs, and test for the docs