4 files changed, 14 insertions, 1 deletions

diff --git a/enigma/src/main/java/cuchaz/enigma/analysis/ClassInheritanceTreeNode.java b/enigma/src/main/java/cuchaz/enigma/analysis/ClassInheritanceTreeNode.java
index 7904c5f..788c534 100644
--- a/enigma/src/main/java/cuchaz/enigma/analysis/ClassInheritanceTreeNode.java
+++ b/enigma/src/main/java/cuchaz/enigma/analysis/ClassInheritanceTreeNode.java
@@ -44,6 +44,13 @@ public class ClassInheritanceTreeNode extends DefaultMutableTreeNode {
                 return null;
         }
 
+        /**
+         * Returns the class entry represented by this tree node.
+         */
+        public ClassEntry getClassEntry() {
+                return this.obfClassEntry;
+        }
+
         public String getObfClassName() {
                 return this.obfClassEntry.getFullName();
         }
diff --git a/enigma/src/main/java/cuchaz/enigma/analysis/MethodInheritanceTreeNode.java b/enigma/src/main/java/cuchaz/enigma/analysis/MethodInheritanceTreeNode.java
index e77b5cc..7bee4dc 100644
--- a/enigma/src/main/java/cuchaz/enigma/analysis/MethodInheritanceTreeNode.java
+++ b/enigma/src/main/java/cuchaz/enigma/analysis/MethodInheritanceTreeNode.java
@@ -48,6 +48,9 @@ public class MethodInheritanceTreeNode extends DefaultMutableTreeNode {
                 return null;
         }
 
+        /**
+         * Returns the method entry represented by this tree node.
+         */
         public MethodEntry getMethodEntry() {
                 return this.entry;
         }
diff --git a/enigma/src/main/java/cuchaz/enigma/analysis/ReferenceTreeNode.java b/enigma/src/main/java/cuchaz/enigma/analysis/ReferenceTreeNode.java
index c0a3a75..ce23cb6 100644
--- a/enigma/src/main/java/cuchaz/enigma/analysis/ReferenceTreeNode.java
+++ b/enigma/src/main/java/cuchaz/enigma/analysis/ReferenceTreeNode.java
@@ -14,6 +14,9 @@ package cuchaz.enigma.analysis;
 import cuchaz.enigma.translation.representation.entry.Entry;
 
 public interface ReferenceTreeNode<E extends Entry<?>, C extends Entry<?>> {
+        /**
+         * Returns the entry represented by this tree node.
+         */
         E getEntry();
 
         EntryReference<E, C> getReference();
diff --git a/enigma/src/main/java/cuchaz/enigma/analysis/StructureTreeNode.java b/enigma/src/main/java/cuchaz/enigma/analysis/StructureTreeNode.java
index 8782b8f..55d75cf 100644
--- a/enigma/src/main/java/cuchaz/enigma/analysis/StructureTreeNode.java
+++ b/enigma/src/main/java/cuchaz/enigma/analysis/StructureTreeNode.java
@@ -26,7 +26,7 @@ public class StructureTreeNode extends DefaultMutableTreeNode {
     }
 
     /**
-     * Returns the parented entry corresponding to this tree node.
+     * Returns the parented entry represented by this tree node.
      */
     public ParentedEntry getEntry() {
         return this.entry;