Fixing JTree issues on click