Update argument handling