Adds type hints for callbacks