diff --git a/README.md b/README.md
index b442201..7060627 100644
--- a/README.md
+++ b/README.md
@@ -10,6 +10,7 @@
 [![Testing](https://github.com/materialsvirtuallab/matcalc/workflows/Testing/badge.svg)](https://github.com/materialsvirtuallab/matcalc/workflows/Testing/badge.svg)
 [![codecov](https://codecov.io/gh/materialsvirtuallab/matcalc/branch/main/graph/badge.svg?token=OR7Z9WWRRC)](https://codecov.io/gh/materialsvirtuallab/matcalc)
 [![Requires Python 3.8+](https://img.shields.io/badge/Python-3.8+-blue.svg?logo=python&logoColor=white)](https://python.org/downloads)
+[![PyPI](https://img.shields.io/pypi/v/matcalc?logo=pypi&logoColor=white)](https://pypi.org/project/matcalc?logo=pypi&logoColor=white)
 </h4>
 
 ## Introduction
diff --git a/setup.py b/setup.py
index 1dbc7a5..7beba61 100644
--- a/setup.py
+++ b/setup.py
@@ -29,6 +29,7 @@
     python_requires=">=3.8",
     package_data={},
     install_requires=("ase", "pymatgen", "joblib", "phonopy"),
+    extras_require={"models": ["matgl", "chgnet"]},
     classifiers=[
         "Development Status :: 4 - Beta",
         "Intended Audience :: Science/Research",