diff --git a/inst/apps/data_analysis/rsconnect/shinyapps.io/agdamsbo/webResearch.dcf b/inst/apps/data_analysis/rsconnect/shinyapps.io/agdamsbo/webResearch.dcf
index 4f4b018..fd874dc 100644
--- a/inst/apps/data_analysis/rsconnect/shinyapps.io/agdamsbo/webResearch.dcf
+++ b/inst/apps/data_analysis/rsconnect/shinyapps.io/agdamsbo/webResearch.dcf
@@ -5,6 +5,6 @@ account: agdamsbo
 server: shinyapps.io
 hostUrl: https://api.shinyapps.io/v1
 appId: 13276335
-bundleId: 9412301
+bundleId: 9425248
 url: https://agdamsbo.shinyapps.io/webResearch/
 version: 1
diff --git a/inst/apps/data_analysis/server.R b/inst/apps/data_analysis/server.R
index 62e2335..c2fe544 100644
--- a/inst/apps/data_analysis/server.R
+++ b/inst/apps/data_analysis/server.R
@@ -84,6 +84,16 @@ server <- function(input, output, session) {
     )
   })
 
+  output$strat_var <- shiny::renderUI({
+    selectInput(
+      inputId = "strat_var",
+      selected = "none",
+      label = "Select variable to stratify baseline",
+      choices = c("none" ,colnames(ds())),
+      multiple = FALSE
+    )
+  })
+
   output$factor_vars <- shiny::renderUI({
     selectizeInput(
       inputId = "factor_vars",
@@ -117,8 +127,8 @@ server <- function(input, output, session) {
 
       data <- data |> factorize(vars = input$factor_vars)
 
-      if (is.factor(data[[input$outcome_var]])) {
-        by.var <- input$outcome_var
+      if (is.factor(data[[input$strat_var]])) {
+        by.var <- input$strat_var
       } else {
         by.var <- NULL
       }
@@ -149,7 +159,14 @@ server <- function(input, output, session) {
               list(
                 by = by.var
               )
-          ),
+          ) |>
+          (\(.x){
+            if (!is.null(by.var)){
+              .x |> gtsummary::add_overall()
+            } else {
+              .x
+            }
+          })(),
         table2 = model |>
           regression_table()
       )
diff --git a/inst/apps/data_analysis/testing_output.rds b/inst/apps/data_analysis/testing_output.rds
deleted file mode 100644
index 9ad95e0..0000000
Binary files a/inst/apps/data_analysis/testing_output.rds and /dev/null differ
diff --git a/inst/apps/data_analysis/ui.R b/inst/apps/data_analysis/ui.R
index 9d892d5..0554818 100644
--- a/inst/apps/data_analysis/ui.R
+++ b/inst/apps/data_analysis/ui.R
@@ -128,6 +128,7 @@ ui <- bslib::page(
         ),
         shiny::helpText(em("Please specify relevant columns from your data, and press 'Load data'")),
         shiny::uiOutput("outcome_var"),
+        shiny::uiOutput("strat_var"),
         shiny::radioButtons(
           inputId = "all",
           label = "Specify covariables",