ui improvements

1 files changed, 12 insertions, 0 deletions

diff --git a/src/cuchaz/enigma/gui/ClassSelectorClassNode.java b/src/cuchaz/enigma/gui/ClassSelectorClassNode.java
index b3d7b89..f054188 100644
--- a/src/cuchaz/enigma/gui/ClassSelectorClassNode.java
+++ b/src/cuchaz/enigma/gui/ClassSelectorClassNode.java
@@ -35,4 +35,16 @@ public class ClassSelectorClassNode extends DefaultMutableTreeNode {
                 }
                 return m_classEntry.getSimpleName();
         }
+        
+        @Override
+        public boolean equals(Object other) {
+                if (other instanceof ClassSelectorClassNode) {
+                        return equals((ClassSelectorClassNode)other);
+                }
+                return false;
+        }
+        
+        public boolean equals(ClassSelectorClassNode other) {
+                return m_classEntry.equals(other.m_classEntry);
+        }
 }