Handle function return types