diff --git a/docker-exec.sh b/docker-exec.sh
index ba77aec..04f53da 100755
--- a/docker-exec.sh
+++ b/docker-exec.sh
@@ -1,3 +1,11 @@
 #!/bin/bash
 
-docker exec -it toolchain-toolchain-1 /bin/zsh
\ No newline at end of file
+#!/bin/bash
+
+# Ensure a command argument is provided
+if [ -z "$1" ]; then
+  echo "Usage: $0 <command>"
+  exit 1
+fi
+
+docker exec -it toolchain-toolchain-1 "$1"
\ No newline at end of file