updated docs and figures