Updated docs to reflect new API