Update docs to reflect new API