diff --git a/client/src/tui/functions/common/login.rs b/client/src/tui/functions/common/login.rs
index 96d3fbb..052d7cf 100644
--- a/client/src/tui/functions/common/login.rs
+++ b/client/src/tui/functions/common/login.rs
@@ -45,16 +45,13 @@ pub async fn save(
             );
 
             // Use the helper method to configure and show the dialog
+            // TODO Implement logic for pressing menu or exit buttons, not imeplementing it now,
+            // need to do other more important stuff now"
             app_state.show_dialog(
                 "Login Success",
                 &success_message,
-                vec!["OK".to_string()], // Pass buttons here
+                vec!["Menu".to_string(), "Exit".to_string()],
             );
-            // REMOVE these lines:
-            // app_state.ui.dialog.dialog_title = "Login Success".to_string();
-            // app_state.ui.dialog.dialog_message = success_message;
-            // app_state.ui.dialog.dialog_show = true;
-            // app_state.ui.dialog.dialog_button_active = true;
 
             Ok("Login successful, details shown in dialog.".to_string())
         }