diff --git a/src/agents/dungeon_master.py b/src/agents/dungeon_master.py
index 261b9fd..6a87c03 100644
--- a/src/agents/dungeon_master.py
+++ b/src/agents/dungeon_master.py
@@ -5,7 +5,7 @@
 from langchain.agents import initialize_agent
 import logging
 from src.npc_generation import generate_npc
-from src.referee import decide_difficulty
+from src.outcome_engine.referee import decide_difficulty
 from src.text_generation.text_generator import get_default_text_generator
 
 # Set up logging
@@ -67,7 +67,7 @@ def narrate(prompt: str) -> str:
     agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION, 
     verbose=False, 
     memory=memory,
-    max_iterations=1,
+    max_iterations=2,
     )
 
 def run_dungeon_master(prompt) -> str:
diff --git a/src/referee.py b/src/outcome_engine/referee.py
similarity index 100%
rename from src/referee.py
rename to src/outcome_engine/referee.py
diff --git a/tests/test_referee.py b/tests/test_referee.py
index cb49347..c0a149b 100644
--- a/tests/test_referee.py
+++ b/tests/test_referee.py
@@ -1,5 +1,5 @@
 import pytest
-from src.referee import Difficulty, decide_difficulty
+from src.outcome_engine.referee import Difficulty, decide_difficulty
 
 @pytest.fixture
 def TRIVIAL_CONTEXT():