update docs and tests