fixed docs, type hints and imports