Updated docs to reflect changes in the API