Updated docs to new API