diff --git a/README.md b/README.md
index 4c146281..4499ae1d 100644
--- a/README.md
+++ b/README.md
@@ -256,7 +256,7 @@ would save data to `gsm8k.jsonl` and also generate a corresponding `gsm8k.html`
 python gsm8k_server.py process --env.data_path_to_save_groups gsm8k.jsonl \
   --openai.base_url https://generativelanguage.googleapis.com/v1beta/openai/ \
   --openai.api_key $GEMINI_API_KEY \
-  --openai.model_name gemini-2.0-flash  
+  --openai.model_name gemini-2.0-flash
 ```
 
 See `python gsm8k_server.py process --help` for more detailed usage.
diff --git a/environments/infinimath/infinimath_env.py b/environments/infinimath/infinimath_env.py
index 26710c0c..6a4d2eb7 100644
--- a/environments/infinimath/infinimath_env.py
+++ b/environments/infinimath/infinimath_env.py
@@ -9,7 +9,12 @@ from typing import Dict, List, Optional, Tuple, Union
 from dotenv import load_dotenv
 from openai import AsyncOpenAI
 
-from atroposlib.envs.base import BaseEnv, BaseEnvConfig, APIServerConfig, ScoredDataGroup
+from atroposlib.envs.base import (
+    APIServerConfig,
+    BaseEnv,
+    BaseEnvConfig,
+    ScoredDataGroup,
+)
 from atroposlib.utils.tokenize_for_trainer import tokenize_for_trainer
 
 from .curriculum import MathCurriculum