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