update docs and imports