public void actionPerformed(ActionEvent e) {

    Object source = e.getSource();

    if (source == cancelButton) {
      orUIManager.cancelUpgrade();
    } else if (source == doneButton) {
      orUIManager.executeUpgrade();
    }
  }