Add generation select support to UI