fixed docs, type hints and imports
Merge pull request #558 from gharlan/fixed-docs-typehints-imports

fixed docs, type hints and imports