diff --git a/environments/community/philosophical_rlaif_env.py b/environments/community/philosophical_rlaif_env.py
index b07df44d..04d98266 100644
--- a/environments/community/philosophical_rlaif_env.py
+++ b/environments/community/philosophical_rlaif_env.py
@@ -3,10 +3,10 @@ import random
 import re  # Added import for regular expressions
 from typing import Dict, List, Optional, Tuple
 
+import wandb  # type: ignore
 from datasets import load_dataset  # type: ignore
 from pydantic import Field
 
-import wandb  # type: ignore
 from atroposlib.envs.base import BaseEnvConfig  # Ensure BaseEnvConfig is imported
 from atroposlib.envs.base import (
     APIServerConfig,
diff --git a/environments/community/ufc_prediction_env/ufc_server.py b/environments/community/ufc_prediction_env/ufc_server.py
index 2e654852..f127449d 100644
--- a/environments/community/ufc_prediction_env/ufc_server.py
+++ b/environments/community/ufc_prediction_env/ufc_server.py
@@ -88,21 +88,21 @@ class UFCEnv(BaseEnv):
                     return f"{name}: (No stats available)"
                 return (
                     f"Name: {name}\n"
-                    f"Wins: {stats.get('wins','?')}  "
-                    f"Losses: {stats.get('losses','?')}  "
-                    f"Age: {stats.get('age','?')}\n"
-                    f"Height: {stats.get('height','?')} cm  "
-                    f"Weight: {stats.get('weight','?')} kg  "
-                    f"Reach: {stats.get('reach','?')} cm  "
-                    f"Stance: {stats.get('stance','?')}\n"
-                    f"SLpM: {stats.get('SLpM','?')}  "
-                    f"Sig Str Acc: {stats.get('sig_str_acc','?')}  "
-                    f"SApM: {stats.get('SApM','?')}  "
-                    f"Str Def: {stats.get('str_def','?')}\n"
-                    f"TD Avg: {stats.get('td_avg','?')}  "
-                    f"TD Acc: {stats.get('td_acc','?')}  "
-                    f"TD Def: {stats.get('td_def','?')}  "
-                    f"Sub Avg: {stats.get('sub_avg','?')}\n"
+                    f"Wins: {stats.get('wins', '?')}  "
+                    f"Losses: {stats.get('losses', '?')}  "
+                    f"Age: {stats.get('age', '?')}\n"
+                    f"Height: {stats.get('height', '?')} cm  "
+                    f"Weight: {stats.get('weight', '?')} kg  "
+                    f"Reach: {stats.get('reach', '?')} cm  "
+                    f"Stance: {stats.get('stance', '?')}\n"
+                    f"SLpM: {stats.get('SLpM', '?')}  "
+                    f"Sig Str Acc: {stats.get('sig_str_acc', '?')}  "
+                    f"SApM: {stats.get('SApM', '?')}  "
+                    f"Str Def: {stats.get('str_def', '?')}\n"
+                    f"TD Avg: {stats.get('td_avg', '?')}  "
+                    f"TD Acc: {stats.get('td_acc', '?')}  "
+                    f"TD Def: {stats.get('td_def', '?')}  "
+                    f"Sub Avg: {stats.get('sub_avg', '?')}\n"
                 )
 
             prompt_text = (
diff --git a/environments/math_server.py b/environments/math_server.py
index aafcefc5..8235f0e5 100644
--- a/environments/math_server.py
+++ b/environments/math_server.py
@@ -5,6 +5,7 @@ from concurrent.futures import ProcessPoolExecutor
 from difflib import SequenceMatcher
 from typing import Dict, List, Optional, Tuple
 
+import wandb
 from datasets import load_dataset
 from latex2sympy2_extended import NormalizationConfig
 from math_verify import LatexExtractionConfig, parse, verify
@@ -12,7 +13,6 @@ from math_verify.errors import TimeoutException
 from pydantic import Field
 from tqdm.asyncio import tqdm_asyncio
 
-import wandb
 from atroposlib.envs.base import (
     APIServerConfig,
     BaseEnv,
diff --git a/environments/math_server_zero.py b/environments/math_server_zero.py
index 17e81ccf..bee1c37e 100644
--- a/environments/math_server_zero.py
+++ b/environments/math_server_zero.py
@@ -9,6 +9,7 @@ import re
 from concurrent.futures import ProcessPoolExecutor
 from typing import Dict, List, Optional, Tuple
 
+import wandb
 from datasets import load_dataset
 from latex2sympy2_extended import NormalizationConfig
 from math_verify import LatexExtractionConfig, parse, verify
@@ -16,7 +17,6 @@ from math_verify.errors import TimeoutException
 from pydantic import Field
 from tqdm.asyncio import tqdm_asyncio
 
-import wandb
 from atroposlib.envs.base import (
     BaseEnv,
     BaseEnvConfig,
diff --git a/environments/tool_calling_server.py b/environments/tool_calling_server.py
index a6d8633b..7de921aa 100644
--- a/environments/tool_calling_server.py
+++ b/environments/tool_calling_server.py
@@ -3,10 +3,10 @@ import random
 import re
 from typing import Dict, List, Optional, Tuple, Union
 
+import wandb
 from datasets import load_dataset
 from tqdm.asyncio import tqdm_asyncio
 
-import wandb
 from atroposlib.envs.base import (
     APIServerConfig,
     BaseEnv,