diff --git a/vmcreate1.sh b/vmcreate1.sh
index b85db27..9c1aee8 100644
--- a/vmcreate1.sh
+++ b/vmcreate1.sh
@@ -79,14 +79,16 @@ ip_check() {
 }
 
 #Проверяем ключ
-if [ -v privkey ]; then
-    if [[ ! -f privkey ]]; then 
+if [ -v pubvkey ]; then
+    if [[ ! -f pubvkey ]]; then 
         input=y
         echo "Private key does not exist. Generate new pair? y/n"
         read input
         case $input in
             y) keypair_generate;;
             Y) keypair_generate;;
+            n) read -p "Enter name for you private key: " privkey && keypair_generate ("$privkey");;
+            N) read -p "Enter name for you private key: " privkey && keypair_generate ("$privkey");;
             *) echo "Use '-f' flag for help"; exit 0;;
         esac
         else keypair_generate
@@ -97,9 +99,9 @@ if [ -v privkey ]; then
 fi
 
 keypair_generate (){
-ssh-keygen -t rsa -N "" -f key.pub && 
+ssh-keygen -t rsa -N "" -f $1 
 #указать в конце функции переменную для вывода названия приватного ключа
-privkey=
+pubkey="$privkey".pub
 }
 
 embed_key(){