diff --git a/.github/workflows/py-tests.yaml b/.github/workflows/py-tests.yaml
index ab29bbd6..0e46f531 100644
--- a/.github/workflows/py-tests.yaml
+++ b/.github/workflows/py-tests.yaml
@@ -26,7 +26,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ['3.9', '3.10']
+        python-version: ['3.9', '3.10', '3.11']
 
     env:
       PYTHON: ${{ matrix.python-version }}
@@ -44,4 +44,4 @@ jobs:
       run: |
         pip install maturin pytest
         pip install -e ".[dev]" 
-        pytest python/altrios -s
\ No newline at end of file
+        pytest python/altrios -s
diff --git a/.github/workflows/wheels.yaml b/.github/workflows/wheels.yaml
index 310a68f5..2e23a889 100644
--- a/.github/workflows/wheels.yaml
+++ b/.github/workflows/wheels.yaml
@@ -23,6 +23,7 @@ jobs:
         python-version:
           - "9"
           - "10"
+          - "11"
         include:
           - os: ubuntu
             platform: linux
diff --git a/pyproject.toml b/pyproject.toml
index 0034ef7b..c6083162 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -24,7 +24,7 @@ authors = [
 description = "Tool for modeling and optimization of advanced locomotive powertrains for freight rail decarbonization."
 readme = "README.md"
 license = { file = "LICENSE.md" }
-requires-python = ">=3.9, <3.11"
+requires-python = ">=3.9, <3.12"
 classifiers = [
     "Programming Language :: Python :: 3",
     "License :: OSI Approved :: BSD License",
diff --git a/python/altrios/demos/sim_manager_demo.py b/python/altrios/demos/sim_manager_demo.py
index aed09db4..65d99b37 100644
--- a/python/altrios/demos/sim_manager_demo.py
+++ b/python/altrios/demos/sim_manager_demo.py
@@ -180,3 +180,5 @@
         plt.tight_layout()
         plt.savefig(plot_dir / f"sim num {idx}.png")
         plt.savefig(plot_dir / f"sim num {idx}.svg")
+
+# %%
diff --git a/python/altrios/train_planner.py b/python/altrios/train_planner.py
index e8165e6f..f53f38a2 100644
--- a/python/altrios/train_planner.py
+++ b/python/altrios/train_planner.py
@@ -1,4 +1,4 @@
-from pathlib import Path
+jfrom pathlib import Path
 from typing import Union
 import numpy as np
 from scipy.stats import rankdata