update docs and add to return object