Add type hints