Updated docs to reflect recent API changes