Updates to the docs and API