Should add id_generator to the list of args, since this function has a list of args.