diff --git a/home/icons/comfyui.png b/home/icons/comfyui.png
new file mode 100644
index 0000000..d8a890d
Binary files /dev/null and b/home/icons/comfyui.png differ
diff --git a/home/icons/openwebui.png b/home/icons/openwebui.png
new file mode 100644
index 0000000..389196c
Binary files /dev/null and b/home/icons/openwebui.png differ
diff --git a/home/programs/ollama.nix b/home/programs/ollama.nix
index edb88bb..63866da 100644
--- a/home/programs/ollama.nix
+++ b/home/programs/ollama.nix
@@ -20,8 +20,10 @@
       Service = let
         docker = lib.getExe pkgs.docker;
       in {
-        ExecStartPre = "bash -c '${docker} inspect open-webui || ${docker} create -e PORT=3021 --network host --gpus all -v ollama:/root/.ollama -v open-webui:/app/backend/data --name open-webui --restart always ghcr.io/open-webui/open-webui:ollama'";
+        Type = "exec";
+        ExecStartPre = "-${docker} create -e PORT=3021 --network host --gpus all -v ollama:/root/.ollama -v open-webui:/app/backend/data --name open-webui --restart always ghcr.io/open-webui/open-webui:ollama";
         ExecStart = "${docker} start open-webui";
+        ExecStop = "${docker} stop open-webui";
       };
     };
   };
@@ -30,6 +32,7 @@
     open-webui = {
       name = "Ollama OpenWebUI";
       type = "Application";
+      icon = ../icons/openwebui.png;
       exec = "${let
         notify-send = "${lib.getExe' pkgs.libnotify "notify-send"}";
         systemctl = "${lib.getExe' pkgs.systemd "systemctl"}";
diff --git a/home/programs/stable-diffusion.nix b/home/programs/stable-diffusion.nix
index bc10c19..c6dbc97 100644
--- a/home/programs/stable-diffusion.nix
+++ b/home/programs/stable-diffusion.nix
@@ -32,6 +32,7 @@
     stable-diffusion = {
       name = "Stable Diffusion";
       type = "Application";
+      icon = ../icons/comfyui.png;
       exec = "${let
         notify-send = "${lib.getExe' pkgs.libnotify "notify-send"}";
         systemctl = "${lib.getExe' pkgs.systemd "systemctl"}";