diff --git a/go-demo-4/main.go b/go-demo-4/main.go
index fe05c5d..14b0041 100644
--- a/go-demo-4/main.go
+++ b/go-demo-4/main.go
@@ -9,6 +9,12 @@ import (
 	"github.com/fatih/color"
 )
 
+var menu = map[string]func(*account.VaultWithDb){
+	"1": createAccount,
+	"2": findAccount,
+	"3": deleteAccount,
+}
+
 func main() {
 	// fmt.Println(output.Sum(100, 42))
 	// fmt.Println(output.Sum(10.7, 3.14))
@@ -24,16 +30,21 @@ Menu:
 			"4. Выход",
 			"Выберите вариант",
 		})
-		switch variant {
-		case "1":
-			createAccount(vault)
-		case "2":
-			findAccount(vault)
-		case "3":
-			deleteAccount(vault)
-		default:
+		menuFunc := menu[variant]
+		if menuFunc == nil {
 			break Menu
 		}
+		menuFunc(vault)
+		// switch variant {
+		// case "1":
+		// 	createAccount(vault)
+		// case "2":
+		// 	findAccount(vault)
+		// case "3":
+		// 	deleteAccount(vault)
+		// default:
+		// 	break Menu
+		// }
 	}
 }