diff --git a/scripts/bootstrap.sh b/scripts/bootstrap.sh
index a1f43fa..d1bd2ac 100644
--- a/scripts/bootstrap.sh
+++ b/scripts/bootstrap.sh
@@ -16,7 +16,11 @@ mkdir -p "/root/venvs"
 
 # Clone SIA repository
 echo "Cloning SIA repository..."
-git clone "$SIA_REPO_URL" "/root/sia"
+[ ! -d "/root/sia" ] && git clone "$SIA_REPO_URL" "/root/sia"
+
+# Fixing permissions
+echo "Fixing permissions..."
+chmod +x /root/sia/scripts/*.sh
 
 # Install Node.js if needed
 if ! command -v node &> /dev/null; then
@@ -35,7 +39,6 @@ npm run build
 ln -s "/root/sia/web/dist" "/root/static"
 
 # Install SIA dependencies
-echo "Installing SIA dependencies..."
 "/root/sia/scripts/install.sh"
 
 # Finetune model
diff --git a/scripts/deploy.sh b/scripts/deploy.sh
index 0ed7bb5..d0f600d 100644
--- a/scripts/deploy.sh
+++ b/scripts/deploy.sh
@@ -8,6 +8,7 @@ export MSYS_NO_PATHCONV=1
 
 # =================== Configuration ===================
 # These can be overridden by environment variables or .env file
+SIA_REPO_URL=${SIA_REPO_URL:-"git@git.nielsgeens.be:2222/llm/SIA.git"}
 
 # Pod configuration
 GPU_TYPE=${GPU_TYPE:-"NVIDIA GeForce RTX 3090"}
@@ -41,7 +42,7 @@ output=$(runpodctl create pod \
     --mem $MEMORY \
     --vcpu $CPU_COUNT \
     --ports "8080/http,8888/http" \
-    --env "SIA_REPO_URL=git.nielsgeens.be/llm/SIA" \
+    --env "SIA_REPO_URL=$SIA_REPO_URL" \
     --env "SIA_GITHUB_API_KEY=$SIA_GITHUB_API_KEY" \
     --env "SIA_GITEA_API_KEY=$SIA_GITEA_API_KEY" \
     --env "SIA_HF_API_KEY=$SIA_HF_API_KEY" \
diff --git a/tools/train/bin/train b/tools/train/bin/train
index 1fd21b7..868254b 100644
--- a/tools/train/bin/train
+++ b/tools/train/bin/train
@@ -12,4 +12,4 @@ fi
 
 mkdir -p "$OUTPUT_DIR"
 
-/root/venvs/train/bin/python -m train.qwq "$@"
\ No newline at end of file
+/root/venvs/train/bin/python -m train.qwq --output-dir "$OUTPUT_DIR"
\ No newline at end of file