diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 2504afc..97c45bc 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -29,6 +29,10 @@ jobs:
           cache: "poetry"
       - name: Install dependencies
         run: poetry install
+      - name: Install pre-commit
+        run: pip install pre-commit
+      - name: Run pre-commit
+        run: pre-commit run --all-files
       - name: Run unit tests
         run: |
            poetry run pytest .
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 5d0db2e..ebea2e2 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -43,6 +43,10 @@ repos:
     -   id: check-shebang-scripts-are-executable
     -   id: check-added-large-files
         args: ['--maxkb=100']
+    -   id: trailing-whitespace
+        types: [python]
+    -   id: end-of-file-fixer
+        types: [python]
 
   - repo: https://github.com/executablebooks/mdformat
     rev: 0.7.17
diff --git a/agentic_security/static/index.html b/agentic_security/static/index.html
index c1b469f..f0301c9 100644
--- a/agentic_security/static/index.html
+++ b/agentic_security/static/index.html
@@ -437,7 +437,7 @@
                   <th class="p-3">Vulnerability Module</th>
                   <th class="p-3">% Strength</th>
                   <th class="p-3">Number of Tokens</th>
-                  <th class="p-3">Cost (in gpt-3 tokens)</th>
+                  <th class="p-3">Approx Cost (in tokens)</th>
                 </tr>
               </thead>
               <tbody>
diff --git a/mkdocs.yml b/mkdocs.yml
index 9810b67..8186e04 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -8,9 +8,13 @@ repo_name: msoedov/agentic_security
 copyright: Maintained by <a href="https://msoedov.github.io">Agentic Security Team</a>.
 
 nav:
-  - Home: index.md
+  - Adventure starts here:
+      - Overview: index.md
+      - Quickstart: quickstart.md
+      - Design: design.md
+      - Abstractions: abstractions.md
   - Features: probe_data.md
-  - Core Concepts:
+  - Concepts:
     - Probe Actor: probe_actor.md
     - Refusal Actor: refusal_classifier_plugins.md
     - Agent Spec: http_spec.md
@@ -26,10 +30,32 @@ nav:
     - Image Generation: image_generation.md
     - Stenography Functions: stenography.md
     - Reinforcement Learning Optimization: rl_model.md
+  - WIP:
+    - Agent Operator: operator.md
   - Reference:
     - API Reference: api_reference.md
-  - Community:
-    - Contributing: contributing.md
+  # - Project:
+  #     - Setup: setup.md
+  #     - Version control: version_control.md
+  #     - Docker: docker.md
+  #     - Variables: variables.md
+  #     - Custom libraries: custom_libraries.md
+  #     - Database: database.md
+  #     - Credentials: credentials.md
+  #     - Code execution: code_execution.md
+  #     - Settings: settings.md
+  #     - Version upgrades: version_upgrades.md
+  # - Contributing:
+  #     - Overview: contributing_overview.md
+  #     - Dev environment: dev_environment.md
+  #     - Backend: backend.md
+  #     - Frontend: frontend.md
+  #     - Documentation: documentation.md
+  # - About:
+  #     - Code of conduct: code_of_conduct.md
+  #     - Usage statistics: usage_statistics.md
+  #     - FAQ: faq.md
+  #     - Changelog: changelog.md
 
 plugins:
   - search