2 files changed, 10 insertions, 1 deletions

diff --git a/src/main/java/cuchaz/enigma/gui/highlight/BoxHighlightPainter.java b/src/main/java/cuchaz/enigma/gui/highlight/BoxHighlightPainter.java
index 10366ce..cef6494 100644
--- a/src/main/java/cuchaz/enigma/gui/highlight/BoxHighlightPainter.java
+++ b/src/main/java/cuchaz/enigma/gui/highlight/BoxHighlightPainter.java
@@ -34,7 +34,9 @@ public class BoxHighlightPainter implements Highlighter.HighlightPainter {
         public static Rectangle getBounds(JTextComponent text, int start, int end) {
                 try {
                         // determine the bounds of the text
-                        Rectangle bounds = text.modelToView(start).union(text.modelToView(end));
+                        Rectangle startRect = text.modelToView(start);
+                        Rectangle endRect = text.modelToView(end);
+                        Rectangle bounds = startRect.union(endRect);
 
                         // adjust the box so it looks nice
                         bounds.x -= 2;
diff --git a/src/main/java/cuchaz/enigma/gui/highlight/TokenHighlightType.java b/src/main/java/cuchaz/enigma/gui/highlight/TokenHighlightType.java
new file mode 100644
index 0000000..ae23f32
--- /dev/null
+++ b/src/main/java/cuchaz/enigma/gui/highlight/TokenHighlightType.java
@@ -0,0 +1,7 @@
+package cuchaz.enigma.gui.highlight;
+
+public enum TokenHighlightType {
+        OBFUSCATED,
+        DEOBFUSCATED,
+        PROPOSED
+}