public void actionPerformed(ActionEvent e) { Object source = e.getSource(); if (source == cancelButton) { orUIManager.cancelUpgrade(); } else if (source == doneButton) { orUIManager.executeUpgrade(); } }