diff --git a/.github/workflows/build-and-test.yaml b/.github/workflows/build-and-test.yaml
index 55be899..be85d2b 100644
--- a/.github/workflows/build-and-test.yaml
+++ b/.github/workflows/build-and-test.yaml
@@ -12,7 +12,7 @@ jobs:
     strategy:
       matrix:
         os: [ubuntu-latest, windows-latest, macos-latest]
-        python-version: ['3.7', '3.11']
+        python-version: ['3.7', '3.8', '3.9', '3.10', '3.11']
     steps:
     - uses: actions/checkout@v2
 
@@ -24,6 +24,7 @@ jobs:
     - uses: syphar/restore-virtualenv@v1
       id: cache-virtualenv
       with:
+        custom_cache_key_element: ABLkit
         requirement_files: requirements.txt
 
     - uses: syphar/restore-pip-download-cache@v1
diff --git a/.gitignore b/.gitignore
index d041dd7..ba96e9d 100644
--- a/.gitignore
+++ b/.gitignore
@@ -6,7 +6,7 @@ examples/**/*.png
 *.ckpt
 results
 raw/
-*.egg-info/
+ablkit.egg-info/
 examples/**/*.jpg
 .idea/
 build/
diff --git a/README.md b/README.md
index d718c5c..cdf86b5 100644
--- a/README.md
+++ b/README.md
@@ -1,6 +1,6 @@
 <div align="center">
 
-![ABLkit logo](https://raw.githubusercontent.com/AbductiveLearning/ABLkit/main/docs/_static/img/logo.png)
+<img src="https://raw.githubusercontent.com/AbductiveLearning/ABLkit/main/docs/_static/img/logo.png" width="180">
 
 [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/ablkit)](https://pypi.org/project/ablkit/) [![PyPI version](https://badgen.net/pypi/v/ablkit)](https://pypi.org/project/ablkit/) [![Documentation Status](https://readthedocs.org/projects/ablkit/badge/?version=latest)](https://ablkit.readthedocs.io/en/latest/?badge=latest) [![license](https://img.shields.io/github/license/mashape/apistatus.svg?maxAge=2592000)](https://github.com/AbductiveLearning/ABLkit/blob/main/LICENSE) [![flake8 Lint](https://github.com/AbductiveLearning/ABLkit/actions/workflows/lint.yaml/badge.svg)](https://github.com/AbductiveLearning/ABLkit/actions/workflows/lint.yaml) [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black) [![ABLkit-CI](https://github.com/AbductiveLearning/ABLkit/actions/workflows/build-and-test.yaml/badge.svg)](https://github.com/AbductiveLearning/ABLkit/actions/workflows/build-and-test.yaml)
 
diff --git a/docs/_static/img/logo.png b/docs/_static/img/logo.png
index 8353c9d..6bea82a 100644
Binary files a/docs/_static/img/logo.png and b/docs/_static/img/logo.png differ
diff --git a/examples/hwf/main.py b/examples/hwf/main.py
index fbcd53a..6ea0380 100644
--- a/examples/hwf/main.py
+++ b/examples/hwf/main.py
@@ -78,9 +78,9 @@ def main():
         help="number of epochs in each learning loop iteration (default : 3)",
     )
     parser.add_argument(
-        "--label-smoothing", 
-        type=float, 
-        default=0.2, 
+        "--label-smoothing",
+        type=float,
+        default=0.2,
         help="label smoothing in cross entropy loss (default : 0.2)"
     )
     parser.add_argument(
diff --git a/pyproject.toml b/pyproject.toml
index b41797a..887dd04 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ablkit"
-version = "0.2.1"
+version = "0.2.2"
 authors = [
   { name="LAMDA 2024", email = "abductivelearning@gmail.com" },
 ]
@@ -14,7 +14,7 @@ requires-python = ">=3.7.0"
 license = {text = "MIT LICENSE"}
 keywords = ["machine-learning", "framework", "abductive-learning", "neuro-symbolic"]
 classifiers = [
-    "Development Status :: 4 - Beta",
+    "Development Status :: 5 - Production/Stable",
     "Intended Audience :: Science/Research",
     "Intended Audience :: Developers",
     "License :: OSI Approved :: MIT License",