diff --git a/ml/ui.R b/ml/ui.R
index a0ed80d..e3bc10d 100644
--- a/ml/ui.R
+++ b/ml/ui.R
@@ -59,7 +59,7 @@ dashboardPage(skin="purple",
               dashboardBody(
                 useShinyjs(),
                 #plotlyOutput("plot", height = 800),
-                box(plotOutput("plot"), width="100%"),
+                box(plotOutput("plot"), width="100%", title="Baseline Model (all features)"),
                 box(plotOutput("feature_plot"), width="100%", title="Feature Selection (no resampling)"),
                 #  %>% withSpinner(color="#0dc5c1"), # uncomment if you want the spinner
                 #downloadButton("save", "Download Plot"),