Handle type of argument in function