@Override
    public void actionPerformed(ActionEvent e) {
        String command = e.getActionCommand();
        if (command.equals("remove")) {
            removeClass();
        } else {
            addClass();
        }
    }