Refactor set_arguments to use new yay api