Work on the docs. Closes [gh-2]