Added type hints
Added type hints