diff --git a/Home.py b/Home.py
index 036965e..f64a2a2 100644
--- a/Home.py
+++ b/Home.py
@@ -35,6 +35,8 @@
   schemas:
     - module: oncollamaschemav3
       root_class: OncoLlamaModel
+    - module: oncoschema.schema
+      root_class: OncologyModel
   ```
 
 **2. Prepare Prediction Files**
diff --git a/pages/__init__.py b/pages/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/requirements.txt b/requirements.txt
index 82f9139..e5876c0 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,3 +2,4 @@ streamlit>=1.28.0
 pydantic>=2.0.0
 pyyaml>=6.0
 oncollamaschemav3 @ git+https://github.com/londonaicentre/oncollamaschemav3.git
+londonaicentre-oncoschema>=2.0.2
diff --git a/schemas.yaml b/schemas.yaml
index aa99233..7a3ce52 100644
--- a/schemas.yaml
+++ b/schemas.yaml
@@ -8,3 +8,5 @@
 schemas:
   - module: oncollamaschemav3
     root_class: OncoLlamaModel
+  - module: oncoschema.schema
+    root_class: OncologyModel
diff --git a/utils/__init__.py b/utils/__init__.py
new file mode 100644
index 0000000..e69de29