Updated the documentation to reflect API changes