diff --git a/.local/bin/setup-dotfiles b/.local/bin/setup-dotfiles
deleted file mode 100755
index edb68a7..0000000
--- a/.local/bin/setup-dotfiles
+++ /dev/null
@@ -1,25 +0,0 @@
-#!/bin/zsh
-
-printf "Experimental script! Run it anyway? (y/N) "
-read choice
-[ "$choice" = "y" ] || exit 1
-
-[ ! $1 ] && >&2 echo "Usage: setup-dotfiles [PATH]" && exit 1
-
-cd $1 || { >&2 echo "Invalid path" && exit 1 }
-
-git status 2>/dev/null || { >&2 echo "Directory is not a git directory!" && exit 1 }
-
-cat << EOF
-#!/bin/zsh
-
-export DOTFILES="${PWD}"
-
-[ ! \$DOTFILES ] && unset DOTFILES
-EOF
-
-echo ""
-echo "##############################################################"
-echo "# Please replace ~/.config/zsh/dotfiles with the code above. #"
-echo "# TIPS: Run 'setup-dotfiles > ~/.config/zsh/dotfiles'        #"
-echo "##############################################################"
diff --git a/README.md b/README.md
index 8fa1059..f724b47 100644
--- a/README.md
+++ b/README.md
@@ -1,3 +1,4 @@
+<!-- ZI-DOTFILES -->
 <p align="center">
   <a href="https://github.com/null2264/dotfiles"><img width="25%" src="https://github.com/null2264.png"></img></a>
 </p>
@@ -30,7 +31,7 @@ My Setup:
 > [!WARNING]
 > This section is still WIP!
 
-First time setup only: run `setup-dotfiles <dotfiles path> > ~/.config/zsh/dotfiles`, e.g. `setup-dotfiles ~/Build/GIT/dotfiles > ~/.config/zsh/dotfiles`.
+First time setup only: run `./setup-dotfiles` then follow the instruction.
 
 ## Recommended Project/Software
 - TBSM
diff --git a/setup-dotfiles b/setup-dotfiles
new file mode 100755
index 0000000..6f15711
--- /dev/null
+++ b/setup-dotfiles
@@ -0,0 +1,26 @@
+#!/bin/zsh
+
+printf "Experimental script! Run it anyway? (y/N) "
+read choice
+[ "$choice" = "y" ] || exit 1
+
+[ "$(cat README.md | head -n 1)" = "<!-- ZI-DOTFILES -->" ] || {
+	>&2 echo "Current working directory is not the dotfiles is located!";
+	>&2 echo "Please 'cd' to the dotfiles directory before running the script.";
+	exit 1;
+}
+
+info="""
+########################################################################
+# Please replace '~/.config/zsh/include/dotfiles' with the code above. #
+########################################################################
+"""
+
+echo "#!/bin/zsh"
+echo $info
+
+echo """export ZI_DOTFILES="${PWD}"
+
+[ ! \$ZI_DOTFILES ] && unset ZI_DOTFILES"""
+
+echo $info