Add missing type hints for .size