Update docs to new API