Update argument logic