diff --git a/official-templates/better-comfyui/pre_start.sh b/official-templates/better-comfyui/pre_start.sh
index a56a5ac..5e0142a 100644
--- a/official-templates/better-comfyui/pre_start.sh
+++ b/official-templates/better-comfyui/pre_start.sh
@@ -15,6 +15,14 @@ rsync_with_progress() {
     stdbuf -i0 -o0 -e0 rsync -au --info=progress2 "$@" | stdbuf -i0 -o0 -e0 tr '\r' '\n' | stdbuf -i0 -o0 -e0 grep -oP '\d+%|\d+.\d+[mMgG]' | tqdm --bar-format='{l_bar}{bar}' --total=100 --unit='%' > /dev/null
 }
 
+# Check for required commands
+for cmd in stdbuf rsync tr grep tqdm; do
+    if ! command -v $cmd &> /dev/null; then
+        echo "$cmd could not be found, please install it."
+        exit 1
+    fi
+done
+
 print_feedback "Starting ComfyUI setup..."
 
 print_feedback "Syncing virtual environment..."
@@ -36,4 +44,10 @@ cd /workspace/ComfyUI
 
 print_feedback "Starting ComfyUI server..."
 print_feedback "ComfyUI will be available at http://0.0.0.0:3000"
-exec /workspace/venv/bin/python main.py --listen --port 3000
\ No newline at end of file
+
+# Check if CUSTOM_ARGS is set and not empty
+if [ -n "$CUSTOM_ARGS" ]; then
+    exec /workspace/venv/bin/python main.py --listen --port 3000 $CUSTOM_ARGS
+else
+    exec /workspace/venv/bin/python main.py --listen --port 3000
+fi