diff --git a/install/docker b/install/docker
index 4530e37..4a4c5c6 100755
--- a/install/docker
+++ b/install/docker
@@ -10,24 +10,27 @@ doRemove() {
     sudo apt purge -y docker* podman-docker containerd runc
 }
 
-install_() {
-    echo
-    echo "==============================================="
-    echo "Installing Docker"
-    echo "==============================================="
-    echo
-    doRemove
+installRepo() {
     sudo apt install -y ca-certificates lsb-release
     source /etc/os-release
-    codename="$VERSION_CODENAME"
     key="/etc/apt/keyrings/docker.asc"
     arch=$(dpkg --print-architecture)
     sudo install -m 0755 -d /etc/apt/keyrings
     curl -fsSL "https://download.docker.com/linux/ubuntu/gpg" -o "$key"
     sudo chmod a+r $key
-    echo "deb [arch=$arch signed-by=$key] https://download.docker.com/linux/ubuntu $codename stable" \
+    echo "deb [arch=$arch signed-by=$key] https://download.docker.com/linux/ubuntu $VERSION_CODENAME stable" \
         | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null
     sudo apt update
+}
+
+install() {
+    echo
+    echo "==============================================="
+    echo "Installing Docker"
+    echo "==============================================="
+    echo
+    # doRemove
+    installRepo
     sudo apt install -y \
         docker-ce \
         docker-ce-cli \
@@ -53,7 +56,8 @@ upgrade() {
         docker-ce-cli \
         containerd.io \
         docker-buildx-plugin \
-        docker-compose-plugin
+        docker-compose-plugin \
+        docker-model-plugin
     echo
     echo "Finish!"
     echo
@@ -74,5 +78,7 @@ remove() {
 case "$1" in
     u|upgrade|update) upgrade ;;
     r|remove|d|delete|p|purge) remove ;;
+    install-repo) installRepo ;;
+    # remove-repo) removeRepo ;; #TODO
     *) install ;;
 esac