diff --git a/guided_setup.sh b/guided_setup.sh
index f801d63..d53831b 100644
--- a/guided_setup.sh
+++ b/guided_setup.sh
@@ -1,9 +1,11 @@
 #!/usr/bin/bash
 
+sudo apt update
+sudo apt dist-upgrade -y
 sudo apt install -y lvm2 zsh cloud-utils xfsprogs jq hstr ufw
 source <(curl -sk https://git.service.wiseadvice.eu/wiseadvice_public_repos/wa_public_init_repo/raw/branch/main/common_lib.sh)
 
-mainmenu() {
+what_to_do() {
     echo -ne "
 MAIN MENU
 1) Install K8s
@@ -28,13 +30,12 @@ Choose an option:  "
       ;;
     *)
       echo "Wrong option."
-      mainmenu
+      what_to_do
       ;;
     esac
 }
 
-init_repos
-mainmenu
+what_to_do