diff --git a/CHANGELOG.md b/CHANGELOG.md
index 29600c1..d4ff227 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,10 +1,13 @@
+## 1.0.0+2
+- style: fix code style
+
 ## 1.0.0+1
 - doc: images in readme
+
 ## 1.0.0
 **BREAKING: **
   * api request moved to service
   * Doki is singleton now
 
 ## 0.0.1
-
 * Testing upload
diff --git a/README.md b/README.md
index 94c2708..963509c 100644
--- a/README.md
+++ b/README.md
@@ -1,4 +1,10 @@
-<img alt="Doki Dont kill my app!" src="https://raw.githubusercontent.com/zoocityboy/flutter_doki/main/assets/doki.png">
+
+<picture id="github_header" style="display:block;">
+  <source media="(prefers-color-scheme: dark)" srcset="https://raw.githubusercontent.com/zoocityboy/flutter_doki/main/assets/doki_dark.png">
+  <img alt="Doki Dont kill my app!" src="https://raw.githubusercontent.com/zoocityboy/flutter_doki/main/assets/doki.png">
+</picture>
+<img id="pubdev_header" style="display:none;" alt="Doki Dont kill my app!" src="https://raw.githubusercontent.com/zoocityboy/flutter_doki/main/assets/doki.png">
+
 <br>
 
 Developed by 🦏 [zoocityboy][zoocityboy_link]
@@ -60,11 +66,11 @@ DokiBadgeWidget(
 This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.
 
 -------------------
-
-<picture>
+<picture id="github_zoocityboy" style="display:block;">
   <source media="(prefers-color-scheme: dark)" srcset="https://raw.githubusercontent.com/zoocityboy/zoo_brand/main/styles/README/zoocityboy_light.png">
   <img alt="Flutter developer Zoocityboy" src="https://raw.githubusercontent.com/zoocityboy/zoo_brand/main/styles/README/zoocityboy_dark.png">
 </picture>
+<img id="pubdev_zoocityboy" alt="Flutter developer Zoocityboy" src="https://raw.githubusercontent.com/zoocityboy/zoo_brand/main/styles/README/zoocityboy_dark.png">
 
 
 [logo_black]:https://raw.githubusercontent.com/zoocityboy/zoo_brand/main/styles/README/zoocityboy_dark.png#gh-light-mode-only
diff --git a/tool/publish.sh b/tool/publish.sh
index c6c192a..7bf82c6 100644
--- a/tool/publish.sh
+++ b/tool/publish.sh
@@ -1,6 +1,33 @@
-#!/bin/sh
-mv README.md README.md.bak
-mv README_publish.md README.md
+#!/bin/bash
+
+# Make script exit on first error
+set -e
+
+# Store the script's directory
+SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+PROJECT_DIR="$(dirname "$SCRIPT_DIR")"
+
+echo "📦 Preparing files for publishing..."
+
+# Create temporary README for publishing
+cp "$PROJECT_DIR/README.md" "$PROJECT_DIR/README.md.bak"
+
+# Remove github-specific elements and comments
+sed -i '' \
+    -e '/<!--github-->/,/<!--\^github-->/d' \
+    -e '/<picture id="github_.*>/,/<\/picture>/d' \
+    -e 's/id="pubdev_.*" style="display:none;"/id="pubdev_header"/' \
+    "$PROJECT_DIR/README.md"
+
+# Commit changes
+git commit -am "chore: prepare README.md for publishing."
+
+echo "🚀 Publishing to pub.dev..."
 flutter pub publish
-mv README.md README_publish.md
-mv README.md.bak README.md
\ No newline at end of file
+
+echo "🧹 Cleaning up..."
+# Restore original README
+git reset --hard HEAD^1
+# mv "$PROJECT_DIR/README.md.bak" "$PROJECT_DIR/README.md"
+
+echo "✅ Published successfully!"
\ No newline at end of file