diff --git a/wire_g b/wire_g
index aa76896..f575916 100755
--- a/wire_g
+++ b/wire_g
@@ -156,7 +156,7 @@ remove(){
                     pkill yad
             fi
      fi } 
-export -f remove
+
 
 # function end
 ############################################################################################################ 
@@ -229,7 +229,7 @@ if selection=$(yad --no-klick --list --center \
                    --buttons-layout=center \
                    --button=$"Start:0" --button=$"Stop":"bash -c up_or_down" \
                    --button=$"Import":"bash -c import_wg" \
-                   --button=$"Remove":"bash -c remove" --separator="" \
+                   --button=$"Remove:4" --separator="" \
                    --button=$"About":"bash -c about" \
                    --window-icon=${wg_pic}wg-vpn.png \
                    --borders=8 --image-on-top --image=${wg_pic}wg-active.png \
@@ -250,6 +250,20 @@ if selection=$(yad --no-klick --list --center \
                 connect
         fi
 fi
+ret=$? 
+
+[[ $? -eq 1 ]] && exit 0
+
+if [[ $ret -eq 4 ]]
+    then 
+        remove
+fi
+
+if [[ $ret -eq 5 ]]
+    then 
+        about
+fi
+
 
 ############################################  END  ##########################################################