update docs and little bug fixes