From 5d6668d0385b18378c127842f9b2fd5582209181 Mon Sep 17 00:00:00 2001
From: abunai10 <abunai10@proton.me>
Date: Thu, 24 Aug 2023 13:15:02 -0400
Subject: [PATCH] Update main.lua to support enhancement shaman and arms
 warrior.

---
 dist/main.lua | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/dist/main.lua b/dist/main.lua
index e69de29..a3f5e74 100644
--- a/dist/main.lua
+++ b/dist/main.lua
@@ -0,0 +1,9 @@
+local Unlocker, Caffeine = ...
+
+local _, class, classID = UnitClass("player")
+
+if class == "WARRIOR" then
+    Caffeine:Require('@Warrior/arms')
+elseif class == "SHAMAN" then
+    Caffeine:Require('@Shaman/enhancement')
+end
\ No newline at end of file