diff --git a/config.sh b/config.sh
index b7c7b03f9efce640fb834c8fdef473c4446b07b0..fb04d914f0dc86876b68c225c5f93a250783d272 100644
--- a/config.sh
+++ b/config.sh
@@ -6,38 +6,53 @@ SOURCE_PATH="${BASH_SOURCE[0]:-${(%):-%x}}"
 RELATIVE_PATH="$(dirname "$SOURCE_PATH")"
 ABSOLUTE_PATH="$(realpath "${RELATIVE_PATH}")"
 ####################################
-
+###===================
 ### User Configuration
 YOUR_ENV_NAME="venv"
-
-
 suffix=""
 
 # Check if SLURM_JOB_PARTITION is set
 if [ -n "$SLURM_JOB_PARTITION" ]; then
-    # Check if SLURM_JOB_PARTITION is 'dc-wai'
-    if [ "$SLURM_JOB_PARTITION" = "dc-wai" ]; then
-        suffix="_wai"
-    fi
-    if [ "$SLURM_JOB_PARTITION" = "dc-gh" ]; then
-        suffix="_gh"
-    fi
-
+    case "$SLURM_JOB_PARTITION" in
+        "dc-wai") suffix="_wai" ;;
+        "dc-gh") suffix="_gh" ;;
+        "dc-hwai")
+            # If old "_wai" environment exists, keep the suffix as "_wai", otherwise use "_hwai"
+            if [ -d "${ABSOLUTE_PATH}/${YOUR_ENV_NAME}_${SYSTEMNAME}_wai" ]; then
+                suffix="_wai"
+            else
+                suffix="_hwai"
+            fi
+            ;;
+    esac
+fi
 
-    if [ "$SLURM_JOB_PARTITION" = "dc-hwai" ]; then
-        # From 16.Oct.2024, the dc-wai partiion will be renamed to dc-hwai, to adapt the users who 
-        # already has *_wai venv folder, we need to point the suffix to `wai`
-        # Check if the original '_wai' environment exists
-        if [ -d "${ABSOLUTE_PATH}/${YOUR_ENV_NAME}_${SYSTEMNAME}_wai" ]; then
-            suffix="_wai"
-        else
-            suffix="_hwai"
-        fi
+YOUR_ENV_NAME="${YOUR_ENV_NAME}_$SYSTEMNAME$suffix"
+#==================
+# overwrite $YOUR_ENV_NAME if we want to force acitvate an enviroment
+declare -A FORCE_MAP=(
+    [force_booster]="juwelsbooster"
+    [force_juwels]="juwels"
+    [force_jurecadc]="jurecadc"
+    [force_hwai]="hwai"
+    [force_jedi]="jedi"
+)
+for key in "${!FORCE_MAP[@]}"; do
+    declare "$key=false"
+done
+
+# Parse arguments dynamically
+for ARG in "$@"; do
+    VAR_NAME="${ARG#--}"  # Remove "--" prefix
+    if [[ -n "${FORCE_MAP[$VAR_NAME]}" ]]; then
+        declare "$VAR_NAME=true"
+        YOUR_ENV_NAME="${YOUR_ENV_NAME}_${FORCE_MAP[$VAR_NAME]}"  # Set environment name immediately
+        break  # Stop checking after the first match
     fi
-fi
+done
+
 
 
-YOUR_ENV_NAME="${YOUR_ENV_NAME}_$SYSTEMNAME$suffix"
 
 export ENV_NAME="$(basename "$ABSOLUTE_PATH")"             # Default Name of the venv is the directory that contains this file
 export ENV_DIR="${ABSOLUTE_PATH}/${YOUR_ENV_NAME}"         # Default location of this VENV is "./venv"