Updated documentation to new API. Still needs work