diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 7f68cc0..2b98d97 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -14,7 +14,7 @@ jobs:
     runs-on: "${{ matrix.os }}"
     strategy:
       matrix:
-        python-version: ["3.9", "3.10", "3.11"]
+        python-version: ["3.10", "3.11"]
         os: [ubuntu-latest]
     steps:
     - uses: actions/checkout@v2
diff --git a/.github/workflows/pypi-nightly.yaml b/.github/workflows/pypi-nightly.yaml
index 8fa7a75..5940ed8 100644
--- a/.github/workflows/pypi-nightly.yaml
+++ b/.github/workflows/pypi-nightly.yaml
@@ -28,7 +28,5 @@ jobs:
         TWINE_USERNAME: __token__
         TWINE_PASSWORD: ${{ secrets.PYPI_TOKEN }}
       run: |
-        # Remove the parallel image for Github dark mode in README.md
-        sed -i -e 's$<img src="https://github.com/raw/google/langfun/main/docs/_static/logo.svg#gh-dark-mode-only" width="320px" alt="logo"></img>$$' README.md
         python setup.py sdist bdist_wheel -- --nightly
         twine upload dist/* --skip-existing
\ No newline at end of file
diff --git a/.github/workflows/pypi.yaml b/.github/workflows/pypi.yaml
index aef160b..0b7cfa6 100644
--- a/.github/workflows/pypi.yaml
+++ b/.github/workflows/pypi.yaml
@@ -26,7 +26,5 @@ jobs:
         TWINE_USERNAME: __token__
         TWINE_PASSWORD: ${{ secrets.PYPI_TOKEN }}
       run: |
-        # Remove the parallel image for Github dark mode in README.md
-        sed -i -e 's$<img src="https://github.com/raw/google/langfun/main/docs/_static/logo.svg#gh-dark-mode-only" width="320px" alt="logo"></img>$$' README.md
         python setup.py sdist bdist_wheel
         twine upload dist/*
\ No newline at end of file
diff --git a/README.md b/README.md
index 74b92c7..8b907ce 100644
--- a/README.md
+++ b/README.md
@@ -1,10 +1,17 @@
 <div align="center">
-<img src="https://github.com/raw/google/langfun/main/docs/_static/logo.svg#gh-light-mode-only" width="320px" alt="logo"></img>
-<img src="https://github.com/raw/google/langfun/main/docs/_static/logo.svg#gh-dark-mode-only" width="320px" alt="logo"></img>
+<img src="https://github.com/raw/google/langfun/main/docs/_static/logo.svg" width="520px" alt="logo"></img>
 </div>
 
 # Langfun
 
+[![PyPI version](https://badge.fury.io/py/langfun.svg)](https://badge.fury.io/py/langfun)
+[![codecov](https://codecov.io/gh/google/langfun/branch/main/graph/badge.svg)](https://codecov.io/gh/google/langfun)
+![pytest](https://github.com/google/langfun/actions/workflows/ci.yaml/badge.svg)
+
+[**Installation**](#install) | [**Getting started**](#hello-world)
+
+## What is Langfun
+
 Langfun is a Python library that aims to make language models (LM) fun
 to work with. Its design enables a programming model that flows naturally,
 resembling the human thought process. It emphasizes the reuse and combination of
diff --git a/langfun/__init__.py b/langfun/__init__.py
index c297fc7..64b1d13 100644
--- a/langfun/__init__.py
+++ b/langfun/__init__.py
@@ -39,4 +39,4 @@
 # pylint: enable=g-import-not-at-top
 # pylint: enable=g-bad-import-order
 
-__version__ = "0.0.1"
+__version__ = "0.0.2"
diff --git a/setup.py b/setup.py
index 03fa949..6fb7d4c 100644
--- a/setup.py
+++ b/setup.py
@@ -82,7 +82,6 @@ def _strip_comments_from_line(s: str) -> str:
         'Intended Audience :: Science/Research',
         'License :: OSI Approved :: Apache Software License',
         'Programming Language :: Python :: 3',
-        'Programming Language :: Python :: 3.9',
         'Programming Language :: Python :: 3.10',
         'Programming Language :: Python :: 3.11',
         'Topic :: Scientific/Engineering :: Artificial Intelligence',