1 files changed, 11 insertions, 1 deletions
diff --git a/src/main/java/de/cowtipper/cowlection/data/DataHelper.java b/src/main/java/de/cowtipper/cowlection/data/DataHelper.java
index 37dd4a3..ceae06f 100644
--- a/src/main/java/de/cowtipper/cowlection/data/DataHelper.java
+++ b/src/main/java/de/cowtipper/cowlection/data/DataHelper.java
@@ -1,5 +1,6 @@
 package de.cowtipper.cowlection.data;
 
+import com.google.gson.annotations.SerializedName;
 import de.cowtipper.cowlection.util.Utils;
 import net.minecraft.util.EnumChatFormatting;
 
@@ -92,7 +93,12 @@ public final class DataHelper {
     }
 
     public enum DungeonClass {
-        ARCHER('A'), BERSERK('B'), HEALER('H'), MAGE('M'), TANK('T'), UNKNOWN('U');
+        @SerializedName("archer") ARCHER('A'),
+        @SerializedName("berserk") BERSERK('B'),
+        @SerializedName("healer") HEALER('H'),
+        @SerializedName("mage") MAGE('M'),
+        @SerializedName("tank") TANK('T'),
+        @SerializedName("") UNKNOWN('U');
         private final char shortName;
 
         DungeonClass(char shortName) {
@@ -111,6 +117,10 @@ public final class DataHelper {
         public char getShortName() {
             return shortName;
         }
+
+        public String getName() {
+            return Utils.fancyCase(name());
+        }
     }
 
     public static Map<String, String> getMinions() {