updated docs and tests