Updating docs to reflect the new API