diff --git a/scripts/ui/update_menu.sh b/scripts/ui/update_menu.sh
index c399169..5306836 100755
--- a/scripts/ui/update_menu.sh
+++ b/scripts/ui/update_menu.sh
@@ -1,13 +1,8 @@
 update_ui(){
+  ui_print_versions
   top_border
   echo -e "|     ${green}~~~~~~~~~~~~~~ [ Update Menu ] ~~~~~~~~~~~~~~${default}     | "
   hr
-  #echo -e "|  Check the following website for important software   | "
-  #echo -e "|  changes to the config file before updating Klipper:  | "
-  #echo -e "|                                                       | "
-  #echo -e "|  ${yellow}https://www.klipper3d.org/Config_Changes.html${default}        | "
-  #bottom_border
-  #top_border
   echo -e "|  0) $BB4U_STATUS| "
   hr
   echo -e "|  a) [Update all]       |               |              | "
@@ -32,84 +27,35 @@ update_ui(){
 }
 
 update_menu(){
-  print_header
-  #check system updates and compare component versions
-    ui_print_versions
-  print_msg && clear_msg
   read_bb4u_stat
-  update_ui
+  do_action "" "update_ui"
   while true; do
     read -p "${cyan}Perform action:${default} " action; echo
     case "$action" in
       0)
-        clear
-        print_header
-        toggle_backups
-        print_msg && clear_msg
-        update_ui;;
+        do_action "toggle_backups" "update_ui";;
       1)
-        clear
-        print_header
-        update_klipper && ui_print_versions
-        print_msg && clear_msg
-        update_ui;;
+        do_action "update_klipper" "update_ui";;
       2)
-        clear
-        print_header
-        update_dwc2fk && ui_print_versions
-        print_msg && clear_msg
-        update_ui;;
+        do_action "update_moonraker" "update_ui";;
       3)
-        clear
-        print_header
-        update_dwc2 && ui_print_versions
-        print_msg && clear_msg
-        update_ui;;
+        do_action "update_mainsail" "update_ui";;
       4)
-        clear
-        print_header
-        update_moonraker && ui_print_versions
-        print_msg && clear_msg
-        update_ui;;
+        do_action "update_fluidd" "update_ui";;
       5)
-        clear
-        print_header
-        update_mainsail && ui_print_versions
-        print_msg && clear_msg
-        update_ui;;
+        do_action "update_klipperscreen" "update_ui";;
       6)
-        clear
-        print_header
-        update_fluidd && ui_print_versions
-        print_msg && clear_msg
-        update_ui;;
+        do_action "update_dwc2fk" "update_ui";;
       7)
-        clear
-        print_header
-        update_klipperscreen && ui_print_versions
-        print_msg && clear_msg
-        update_ui;;
+        do_action "update_dwc2" "update_ui";;
       8)
-        clear
-        print_header
-        update_system && ui_print_versions
-        print_msg && clear_msg
-        update_ui;;
+        do_action "update_system" "update_ui";;
       a)
-        clear
-        print_header
-        update_all && ui_print_versions
-        print_msg && clear_msg
-        update_ui;;
+        do_action "update_all" "update_ui";;
       Q|q)
         clear; main_menu; break;;
       *)
-        clear
-        print_header
-        print_unkown_cmd
-        print_msg && clear_msg
-        ui_print_versions
-        update_ui;;
+        deny_action "update_ui";;
     esac
   done
   update_menu