diff --git a/README.md b/README.md
index 2cdea36..b784622 100644
--- a/README.md
+++ b/README.md
@@ -1,10 +1,12 @@
 # KAN-GPT
 
+![PyPI - Downloads](https://img.shields.io/pypi/dd/kan-gpt)
+![PyPI - Version](https://img.shields.io/pypi/v/kan-gpt)
 [![codecov](https://codecov.io/gh/AdityaNG/kan-gpt/branch/main/graph/badge.svg?token=kan-gpt_token_here)](https://codecov.io/gh/AdityaNG/kan-gpt)
 [![CI](https://github.com/AdityaNG/kan-gpt/actions/workflows/main.yml/badge.svg)](https://github.com/AdityaNG/kan-gpt/actions/workflows/main.yml)
-![PyPI - Version](https://img.shields.io/pypi/v/kan-gpt)
 ![GitHub License](https://img.shields.io/github/license/AdityaNG/kan-gpt)
 
+
 The PyTorch implementation of Generative Pre-trained Transformers (GPTs) using Kolmogorov-Arnold Networks (KANs) for language modeling
 
 ## Install it from PyPI
diff --git a/docs/index.md b/docs/index.md
index c9f02fb..2b37bf9 100644
--- a/docs/index.md
+++ b/docs/index.md
@@ -1,8 +1,9 @@
 # KAN-GPT
 
+![PyPI - Downloads](https://img.shields.io/pypi/dd/kan-gpt)
+![PyPI - Version](https://img.shields.io/pypi/v/kan-gpt)
 [![codecov](https://codecov.io/gh/AdityaNG/kan-gpt/branch/main/graph/badge.svg?token=kan-gpt_token_here)](https://codecov.io/gh/AdityaNG/kan-gpt)
 [![CI](https://github.com/AdityaNG/kan-gpt/actions/workflows/main.yml/badge.svg)](https://github.com/AdityaNG/kan-gpt/actions/workflows/main.yml)
-![PyPI - Version](https://img.shields.io/pypi/v/kan-gpt)
 ![GitHub License](https://img.shields.io/github/license/AdityaNG/kan-gpt)
 
 The PyTorch implementation of Generative Pre-trained Transformers (GPTs) using Kolmogorov-Arnold Networks (KANs) for language modeling