diff --git a/src/primaite/notebooks/Data-Manipulation-E2E-Demonstration.ipynb b/src/primaite/notebooks/Data-Manipulation-E2E-Demonstration.ipynb
index a958aa0a..201bf518 100644
--- a/src/primaite/notebooks/Data-Manipulation-E2E-Demonstration.ipynb
+++ b/src/primaite/notebooks/Data-Manipulation-E2E-Demonstration.ipynb
@@ -404,7 +404,7 @@
     "    # don't flatten observations so that we can see what is going on\n",
     "    cfg['agents'][3]['agent_settings']['flatten_obs'] = False\n",
     "\n",
-    "env = PrimaiteGymEnv(game_config = cfg)\n",
+    "env = PrimaiteGymEnv(env_config = cfg)\n",
     "obs, info = env.reset()\n",
     "print('env created successfully')\n",
     "pprint(obs)"
@@ -476,7 +476,9 @@
    "outputs": [],
    "source": [
     "obs, reward, terminated, truncated, info = env.step(9)  # scan database file\n",
+    "print(info)\n",
     "obs, reward, terminated, truncated, info = env.step(1)  # scan webapp service\n",
+    "print(info)\n",
     "pprint(obs['NODES'])"
    ]
   },
@@ -693,7 +695,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.12"
+   "version": "3.10.11"
   }
  },
  "nbformat": 4,
diff --git a/src/primaite/session/environment.py b/src/primaite/session/environment.py
index abbf051b..a34ebf04 100644
--- a/src/primaite/session/environment.py
+++ b/src/primaite/session/environment.py
@@ -58,6 +58,7 @@ class PrimaiteGymEnv(gymnasium.Env):
 
         next_obs = self._get_obs()  # this doesn't update observation, just gets the current observation
         reward = self.agent.reward_function.current_reward
+        _LOGGER.info(f"step: {self.game.step_counter}, Blue reward: {reward}")
         terminated = False
         truncated = self.game.calculate_truncated()
         info = {
diff --git a/src/primaite/session/io.py b/src/primaite/session/io.py
index 22001fd2..75037381 100644
--- a/src/primaite/session/io.py
+++ b/src/primaite/session/io.py
@@ -7,7 +7,7 @@ from pydantic import BaseModel, ConfigDict
 
 from primaite import getLogger, PRIMAITE_PATHS
 from primaite.simulator import LogLevel, SIM_OUTPUT
-from src.primaite.utils.primaite_config_utils import is_dev_mode
+from primaite.utils.primaite_config_utils import is_dev_mode
 
 _LOGGER = getLogger(__name__)