Add imports for type hints