diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -88,16 +88,16 @@ jobs:
       - uses: "actions/setup-python@v4"
         with:
           python-version: "${{ matrix.python-version }}"
-      - name: "Install dependencies (no speedups)"
+      - name: "Install dependencies (no extras)"
         if: matrix.extras == false
         run: |
           python -m pip install --upgrade pip poetry
           poetry install
-      - name: "Install dependencies (with speedups)"
+      - name: "Install dependencies (with extras)"
         if: matrix.extras == true
         run: |
           python -m pip install --upgrade pip poetry
-          poetry install --extras speedups
+          poetry install --all-extras
       - name: "Run tests"
         run: |
           poetry run pytest --cov kasa --cov-report xml

diff --git a/README.md b/README.md
@@ -20,11 +20,19 @@ You can install the most recent release using pip:
 pip install python-kasa
 ```
 
+For enhanced cli tool support (coloring, embedded shell) install with `[shell]`:
+```
+pip install python-kasa[shell]
+```
+
 If you are using cpython, it is recommended to install with `[speedups]` to enable orjson (faster json support):
 ```
 pip install python-kasa[speedups]
 ```
-
+or for both:
+```
+pip install python-kasa[speedups, shell]
+```
 With `[speedups]`, the protocol overhead is roughly an order of magnitude lower (benchmarks available in devtools).
 
 Alternatively, you can clone this repository and use poetry to install the development version:

diff --git a/kasa/tests/test_cli.py b/kasa/tests/test_cli.py
@@ -647,7 +647,7 @@ async def test_errors(mocker):
     assert res.exit_code == 1
     assert (
         "Raised error: Managed to invoke callback without a context object of type 'Device' existing."
-        in res.output
+        in res.output.replace("\n", "")  # Remove newlines from rich formatting
     )
     assert isinstance(res.exception, SystemExit)
 

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -40,8 +40,9 @@ sphinxcontrib-programoutput = { version = "^0", optional = true }
 myst-parser = { version = "*", optional = true }
 docutils = { version = ">=0.17", optional = true }
 
-# shell support
-# ptpython = { version = "*", optional = true }
+# enhanced cli support
+ptpython = { version = "*", optional = true }
+rich = { version = "*", optional = true }
 
 [tool.poetry.group.dev.dependencies]
 pytest = "*"
@@ -60,7 +61,7 @@ coverage = {version = "*", extras = ["toml"]}
 [tool.poetry.extras]
 docs = ["sphinx", "sphinx_rtd_theme", "sphinxcontrib-programoutput", "myst-parser", "docutils"]
 speedups = ["orjson", "kasa-crypt"]
-# shell = ["ptpython"]
+shell = ["ptpython", "rich"]
 
 [tool.coverage.run]
 source = ["kasa"]