diff --git a/Dockerfile b/Dockerfile
index 5a48bcc..290cb83 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -7,9 +7,9 @@ RUN adduser --disabled-password --gecos '' camenduru && \
     adduser camenduru sudo && \
     echo '%sudo ALL=(ALL) NOPASSWD:ALL' >> /etc/sudoers && \
     chown -R camenduru:camenduru /content && \
-    chmod -R 777 /content && \
+    chmod -R 755 /content && \
     chown -R camenduru:camenduru /home && \
-    chmod -R 777 /home && \
+    chmod -R 755 /home && \
     apt update -y && \
     add-apt-repository -y ppa:git-core/ppa && \
     apt update -y && \
@@ -17,23 +17,41 @@ RUN adduser --disabled-password --gecos '' camenduru && \
 
 USER camenduru
 
-# Install Python dependencies, clone ComfyUI, and download necessary models & LoRA
+# Install Python dependencies
 RUN pip install -q opencv-python imageio imageio-ffmpeg ffmpeg-python av runpod \
-    xformers==0.0.25 torchsde==0.2.6 einops==0.8.0 diffusers==0.28.0 transformers==4.41.2 accelerate==0.30.1 && \
-    git clone https://github.com/comfyanonymous/ComfyUI /content/ComfyUI && \
+    xformers==0.0.25 torchsde==0.2.6 einops==0.8.0 diffusers==0.28.0 transformers==4.41.2 accelerate==0.30.1
+
+# Clone the ComfyUI repository
+RUN git clone https://github.com/comfyanonymous/ComfyUI /content/ComfyUI
+
+# Download and place the UNet model
+RUN mkdir -p /content/ComfyUI/models/unet && \
     aria2c --console-log-level=error -c -x 16 -s 16 -k 1M \
-        https://huggingface.co/camenduru/FLUX.1-dev/resolve/main/flux1-dev.sft -d /content/ComfyUI/models/unet -o flux1-dev.sft && \
+        https://huggingface.co/camenduru/FLUX.1-dev/resolve/main/flux1-dev.sft \
+        -d /content/ComfyUI/models/unet -o flux1-dev.sft
+
+# Download and place the CLIP models
+RUN mkdir -p /content/ComfyUI/models/clip && \
     aria2c --console-log-level=error -c -x 16 -s 16 -k 1M \
-        https://huggingface.co/camenduru/FLUX.1-dev/resolve/main/clip_l.safetensors -d /content/ComfyUI/models/clip -o clip_l.safetensors && \
+        https://huggingface.co/camenduru/FLUX.1-dev/resolve/main/clip_l.safetensors \
+        -d /content/ComfyUI/models/clip -o clip_l.safetensors && \
     aria2c --console-log-level=error -c -x 16 -s 16 -k 1M \
-        https://huggingface.co/camenduru/FLUX.1-dev/resolve/main/t5xxl_fp16.safetensors -d /content/ComfyUI/models/clip -o t5xxl_fp16.safetensors && \
+        https://huggingface.co/camenduru/FLUX.1-dev/resolve/main/t5xxl_fp16.safetensors \
+        -d /content/ComfyUI/models/clip -o t5xxl_fp16.safetensors
+
+# Download and place the VAE model
+RUN mkdir -p /content/ComfyUI/models/vae && \
     aria2c --console-log-level=error -c -x 16 -s 16 -k 1M \
-        https://huggingface.co/camenduru/FLUX.1-dev/resolve/main/ae.sft -d /content/ComfyUI/models/vae -o ae.sft && \
-    mkdir -p /content/ComfyUI/models/loras && \
+        https://huggingface.co/camenduru/FLUX.1-dev/resolve/main/ae.sft \
+        -d /content/ComfyUI/models/vae -o ae.sft
+
+# Download and place the LoRa model
+RUN mkdir -p /content/ComfyUI/models/loras && \
     aria2c --console-log-level=error -c -x 16 -s 16 -k 1M \
-        https://civitai.com/api/download/models/896422?type=Model&format=SafeTensor -d /content/ComfyUI/models/loras -o zanshou-kin-flux-ueno-manga-style.safetensors
+        https://civitai.com/api/download/models/896422?type=Model&format=SafeTensor \
+        -d /content/ComfyUI/models/loras -o zanshou-kin-flux-ueno-manga-style.safetensors
 
-# Copy the updated worker script into the container
+# Copy the worker script into the container
 COPY ./worker_runpod.py /content/ComfyUI/worker_runpod.py
 
 WORKDIR /content/ComfyUI