Как реализовать переключатель в JTree?

package TreeTst;
 
import javax.swing.tree.*;
import javax.swing.JTree;
 
 
    /*
    * Новый вид ветки, для изменения отображения текста.
    */
 
 
class FileNode extends DefaultMutableTreeNode {
    
    private boolean showExt;
 
    public boolean isShowExt() {
        return showExt;
    }
 
    public void setShowExt(boolean showExt) {
        this.showExt = showExt;
    }
    
    @Override
    public String toString() {
    if (showExt) return userObject.toString();
    int pointPos = userObject.toString().lastIndexOf('.');
    return userObject.toString().substring(0, pointPos);
    }
    
    public FileNode(String obj) {
    super(obj);
    }
}
 
public class TreeSwitch {
 
    /*
     * Создает модель данных дерева и заполняет ее.
     */
   
    
    public static DefaultTreeModel generateTreeMod() {
    
        DefaultMutableTreeNode rootNode = new DefaultMutableTreeNode("My computer");
        DefaultTreeModel result = new DefaultTreeModel(rootNode);
        
        DefaultMutableTreeNode drvC = new DefaultMutableTreeNode("C:");
        DefaultMutableTreeNode drvD = new DefaultMutableTreeNode("D:");
        
        drvC.add(new DefaultMutableTreeNode("Program files"));
        drvD.add(new FileNode("hello.java"));
        drvD.add(new FileNode("pict.bmp"));
        drvD.add(new FileNode("pron.jpg"));
        drvC.add(new FileNode("pagefile.sys"));
        drvC.add(new FileNode("config.sys"));
        drvC.add(new FileNode("autoexec.bat"));
        rootNode.add(drvC);
        rootNode.add(drvD);
        
        return result;
            
    }
    
    /*
    * Рекурсивно лазит по всем детям узла nod 
    * и делает setShowExt(b), если узел экземпляр FileNode
    */
    
    public static void setShowExt(TreeNode nod, boolean b) {
    if (nod instanceof FileNode) {((FileNode)nod).setShowExt(b); return;} // или1
    //if (nod.isLeaf()) {((FileNode)nod).setShowExt(b); return;} // или2
       //Если или2 - выдает ошибку, попадая на пустую "папку" (Program files)
    int siz = nod.getChildCount();
    for (int i = 0; i<siz; i++) {setShowExt(nod.getChildAt(i),b);}
    }
    
    /*
    * Вызывает setShowExt() для
    * каждого узла 2-го уровня модели дерева(ребенка root)
    */    
    
    public static void setShowExtInLeafs(TreeModel t, boolean b) {
    DefaultMutableTreeNode root;
    root = (DefaultMutableTreeNode)t.getRoot();    
    int siz = root.getChildCount();
    for (int i = 0; i<siz; i++) setShowExt(root.getChildAt(i),b);
    ((DefaultTreeModel)t).reload(); //перерисовать дерево
    }
    
}

import java.awt.*;
import java.awt.event.*;
import javax.swing.JTree;
import javax.swing.tree.*;
import javax.swing.JScrollPane;
import java.util.logging.Level;
import java.util.logging.Logger;
 
public class HtmlTree extends Frame {
 
JTree tree;
BorderLayout bl = new BorderLayout();
Panel pane = new Panel();
Button btOk = new Button("SHOW");
Button btC = new Button("HIDE");
 
private void makeCanClose(){
WindowAdapter a = new WindowAdapter()   
        {
       public void windowClosing(WindowEvent e) { setVisible(false); }
        };    
this.addWindowListener(a);
}
 
private void initWindow(){
setLayout(bl);
setBounds(200,200,300,480);
setVisible(true);
}
 
 
private void initComps(TreeModel tm) {
tree = new JTree(tm);
add(new JScrollPane(tree),BorderLayout.CENTER);
pane.setPreferredSize(new Dimension(300,50));
add(pane,BorderLayout.SOUTH);
pane.setLayout(null);
 
btOk.addMouseListener( new MouseAdapter() {public void mouseClicked(MouseEvent e) {
TreeSwitch.setShowExtInLeafs(tree.getModel(), true);
}});
 
btC.addMouseListener( new MouseAdapter() {public void mouseClicked(MouseEvent e) {
TreeSwitch.setShowExtInLeafs(tree.getModel(), false);
}});
        
pane.add(btOk);
pane.add(btC);
btOk.setBounds(20, 10, 100, 30);
btC.setBounds(140, 10, 100, 30);
}
 
public HtmlTree(TreeModel tm){
makeCanClose();
initWindow();
initComps(tm);
this.revalidate();
}
 
}

public static void main(String[] args) {
HtmlTree tr = new HtmlTree( TreeSwitch.generateTreeMod() );
}