diff --git a/code/Home.py b/code/Home.py
index 34f28b7..bc8310c 100644
--- a/code/Home.py
+++ b/code/Home.py
@@ -207,7 +207,7 @@ def draw_session_plots(df_to_draw_session):
     
     # Setting up layout for each session
     layout_definition = [[1],   # columns in the first row
-                         [1.5, 1],  # columns in the second row
+                         [1, 1],  # columns in the second row
                          [1, 1],
                          ]  
     
@@ -464,10 +464,15 @@ def init():
                                 root='foraging_auto_training/')
     )
     
+    logistic_regression_models = ['Su2022', 'Bari2019', 'Hattori2019', 'Miller2021']
+    
     
     st.session_state.draw_type_mapper_session_level = {'1. Choice history': ('choice_history',   # prefix
                                                             (0, 0),     # location (row_idx, column_idx)
                                                             dict()),
+                                                       **{f'{n + 2}. Logistic regression ({model})': (f'logistic_regression_{model}',   # prefix
+                                                            (1 + int(n/2), n%2),     # location (row_idx, column_idx)
+                                                            dict()) for n, model in enumerate(logistic_regression_models)},
         
                                         # '1. Choice history': ('fitted_choice',   # prefix
                                         #                     (0, 0),     # location (row_idx, column_idx)