Add update argument to script