Add type hints for map.