Changed the method for calling set_args