Add type hint