diff --git a/pwa/src/App.tsx b/pwa/src/App.tsx
index 2f614fd..01a0c8c 100644
--- a/pwa/src/App.tsx
+++ b/pwa/src/App.tsx
@@ -14,6 +14,7 @@ import DiagnosisForm from './components/DiagnosisForm'
 import './scss/App.scss'
 import './antd-mobile.css'
 import DebugShowEasyPeasy from './components/DebugShowEasyPeasy'
+import TreatmentForm from './components/TreatmentForm'
 
 const App: React.FunctionComponent = () => {
     return (
@@ -49,10 +50,19 @@ const App: React.FunctionComponent = () => {
                         navTitle="Diagnosis"
                         leftArrowText="Patient Info"
                         leftArrowRoute="/patientinfo"
-                    >
+                        >
                         <DiagnosisForm />
                     </Wrapper>
-                </Route>
+                    </Route>
+                    <Route path="/treatment">
+                    <Wrapper
+                        navTitle="Treatment"
+                        leftArrowText="Diagnosis"
+                        leftArrowRoute="/diagnosis"
+                        >
+                        <TreatmentForm />
+                    </Wrapper>
+                    </Route>
                     <Route path="/easy-peasy">
                         <DebugShowEasyPeasy />
                     </Route>
diff --git a/pwa/src/components/TreatmentForm.tsx b/pwa/src/components/TreatmentForm.tsx
index 0d1d59d..2731e10 100644
--- a/pwa/src/components/TreatmentForm.tsx
+++ b/pwa/src/components/TreatmentForm.tsx
@@ -13,7 +13,7 @@ export default function TreatmentForm(): ReactElement {
 
     return (
     <>
-        <div>
+        <div id="treatment-form-wrapper" data-testid="treatment-form-wrapper">
             <Accordion defaultActiveKey="0" className="my-accordion">
             <Accordion.Panel header="Multivitamins">
                 <CheckboxItem>Multivitamins - Childrens</CheckboxItem>
@@ -115,9 +115,8 @@ export default function TreatmentForm(): ReactElement {
                     id="comments"/>
             </Accordion.Panel>
             </Accordion>
-        </div>
         
-        <Button onClick={onSubmit} >Submit</Button>
+        </div>
     </>
     )
 }
\ No newline at end of file