Updated documentation with the new API