diff --git a/install.sh b/install.sh
index 9678a65..4fd3562 100644
--- a/install.sh
+++ b/install.sh
@@ -1,52 +1,63 @@
 #!/bin/bash
 
-# 显示菜单
-echo "请选择一个操作:"
-echo "1. 启动容器"
-echo "2. 查看分享链接"
-echo "3. 停止容器"
-echo "4. 更新镜像"
-echo "5. 修改端口" 
-echo "6. 修改Reality目标域名" 
-echo "0. 退出"
+restart_docker(){
+  echo "正在重启容器..."
+  docker-compose -f ./docker-compose.yml down
+  docker-compose -f ./docker-compose.yml up -d
+}
 
-# 读取用户选择
-read -p "输入您的选择: " choice
+main(){
+    # 显示菜单
+    echo "请选择一个操作:"
+    echo "1. 启动容器"
+    echo "2. 查看分享链接"
+    echo "3. 停止容器"
+    echo "4. 更新镜像"
+    echo "5. 修改端口" 
+    echo "6. 修改Reality目标域名" 
+    echo "0. 退出"
 
-# 根据用户选择执行相应的操作
-case $choice in
-    1)
-        # 启动容器
-        echo "启动容器.."
-        bash ./bin/run.sh
-        ;;
-    2)
-        # 查看分享链接
-        echo "查看分享链接.."
-        bash ./bin/print_share_link.sh
-        ;;
-    3)
-        # 停止容器
-        echo "正在停止容器.."
-        docker-compose -f ./docker-compose.yml down
-        ;;
-    4)
-        # 更新镜像
-        bash ./bin/update_docker_images.sh
-        ;;
-    5)
-        # 修改端口
-        bash ./bin/update_port.sh
-        ;;
-    6)
-        # 修改Reality目标域名
-        bash ./bin/update_reality_dest.sh
-        ;;
-    0)
-        # 退出
-        echo "退出程序."
-        ;;
-    *)
-        echo "无效的选择, 请重新选择."
-        ;;
-esac
+    # 读取用户选择
+    read -p "输入您的选择: " choice
+
+    # 根据用户选择执行相应的操作
+    case $choice in
+        1)
+            # 启动容器
+            echo "启动容器.."
+            bash ./bin/run.sh
+            ;;
+        2)
+            # 查看分享链接
+            echo "查看分享链接.."
+            bash ./bin/print_share_link.sh
+            ;;
+        3)
+            # 停止容器
+            echo "正在停止容器.."
+            docker-compose -f ./docker-compose.yml down
+            ;;
+        4)
+            # 更新镜像
+            bash ./bin/update_docker_images.sh
+            ;;
+        5)
+            # 修改端口
+            bash ./bin/update_port.sh
+            restart_docker
+            ;;
+        6)
+            # 修改Reality目标域名
+            bash ./bin/update_reality_dest.sh
+            ;;
+        0)
+            # 退出
+            echo "退出程序."
+            ;;
+        *)
+            echo "无效的选择, 请重新选择."
+            ;;
+    esac
+}
+
+main
\ No newline at end of file