Update doc with the new API