ui improvements

1 files changed, 12 insertions, 0 deletions

diff --git a/src/cuchaz/enigma/gui/ClassSelectorPackageNode.java b/src/cuchaz/enigma/gui/ClassSelectorPackageNode.java
index 451d380..5685abb 100644
--- a/src/cuchaz/enigma/gui/ClassSelectorPackageNode.java
+++ b/src/cuchaz/enigma/gui/ClassSelectorPackageNode.java
@@ -30,4 +30,16 @@ public class ClassSelectorPackageNode extends DefaultMutableTreeNode {
         public String toString() {
                 return m_packageName;
         }
+        
+        @Override
+        public boolean equals(Object other) {
+                if (other instanceof ClassSelectorPackageNode) {
+                        return equals((ClassSelectorPackageNode)other);
+                }
+                return false;
+        }
+        
+        public boolean equals(ClassSelectorPackageNode other) {
+                return m_packageName.equals(other.m_packageName);
+        }
 }