Updated documentation to reflect new API