diff --git a/.update-portainer.sh.swp b/.update-portainer.sh.swp
new file mode 100644
index 0000000..ceffd5b
Binary files /dev/null and b/.update-portainer.sh.swp differ
diff --git a/update-portainer.sh b/update-portainer.sh
index f814201..079d71e 100755
--- a/update-portainer.sh
+++ b/update-portainer.sh
@@ -1,3 +1,47 @@
-docker stop portainer
-docker rm portainer
-docker run -d -p 8000:8000 -p 9000:9000 -p 9443:9443 --name=portainer --restart=always --pull=always -v /var/run/docker.sock:/var/run/docker.sock -v portainer_data:/data portainer/portainer-ee:latest
+#!/bin/bash
+
+function run_io () {
+  docker run -d -p 9001:9001 --name=${image_name} --restart=always -v /var/run/docker.sock:/var/run/docker.sock -v /var/lib/docker/volumes:/var/lib/docker/volumes portainer/${portainer_image}
+
+}
+
+function run_vperanda () {
+  docker run -d -p 8000:8000 -p 9000:9000 -p 9443:9443 --name=${image_name} --restart=always --pull=always -v /var/run/docker.sock:/var/run/docker.sock -v portainer_data:/data portainer/${portainer_image}
+}
+
+portainer_image=""
+image_name=""
+
+case ${HOSTNAME} in
+  vperanda)
+    portainer_image="portainer-ee:latest"
+    image_name="portainer"
+    ;;
+  io)
+    portainer_image="agent:latest"
+    image_name="portainer_agent"
+    ;;
+esac
+
+echo "We're on ${HOSTNAME} so the portainer image is ${portainer_image}. The name will be ${image_name}."
+
+# stop and remove portainer
+docker stop ${image_name}
+docker rm ${image_name}
+
+# run the command to pull and run the latest [portainer,portainer-edge] docker image
+case ${HOSTNAME} in
+  vperanda)
+    run_vperanda
+    ;;
+  io)
+    run_io
+    ;;
+esac
+
+# send a notification to https://notify.peterwood.rocks\lab
+curl \
+  -H priority:default \
+  -H tags:update,docker,portainer,${HOSTNAME} \
+  -d "The ${image_name} portainer image has been updated." \
+  https://notify.peterwood.rocks/lab