Add function: return_args