diff --git a/README.md b/README.md
index ee84795f..578f95ed 100644
--- a/README.md
+++ b/README.md
@@ -3,6 +3,13 @@
 ![Lint Status](https://github.com/assume-framework/assume/actions/workflows/lint-pytest.yaml/badge.svg)
 [![Code Coverage](https://codecov.io/gh/assume-framework/assume/branch/main/graph/badge.svg?token=CZ4FO7P57H)](https://codecov.io/gh/assume-framework/assume)
 [![DOI](https://zenodo.org/badge/DOI/10.5281/zenodo.8088760.svg)](https://doi.org/10.5281/zenodo.8088760)
+
+[![](https://img.shields.io/pypi/v/assume-framework.svg)](https://pypi.org/pypi/assume-framework/)
+[![](https://img.shields.io/pypi/pyversions/assume-framework.svg)](https://pypi.org/pypi/assume-framework/)
+[![](https://img.shields.io/pypi/l/assume-framework.svg)](https://pypi.org/pypi/assume-framework/)
+[![](https://img.shields.io/pypi/status/assume-framework.svg)](https://pypi.org/pypi/assume-framework/)
+[![](https://img.shields.io/readthedocs/assume)](https://assume.readthedocs.io/)
+
 [![Open Tutorials In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1LISiM1QvDIMXU68pJH-NqrMw5w7Awb24?usp=sharing)
 
 **ASSUME** is an open-source toolbox for agent-based simulations of European electricity markets, with a primary focus on the German market setup. Developed as an open-source model, its primary objectives are to ensure usability and customizability for a wide range of users and use cases in the energy system modeling community.
diff --git a/environment_docs.yaml b/environment_docs.yaml
index df747789..91f95eea 100644
--- a/environment_docs.yaml
+++ b/environment_docs.yaml
@@ -1,4 +1,6 @@
 name: assume-doc
+channels:
+  - conda-forge
 dependencies:
 - python>=3.11
 - pip
@@ -7,6 +9,8 @@ dependencies:
 - numpy
 - tqdm
 - SQLAlchemy
+- pytorch
+- pyomo
 
 - pip:
   - .
@@ -15,5 +19,3 @@ dependencies:
   - nbsphinx
   - nbsphinx_link
   - sphinxcontrib-mermaid
-  - pyomo
-  - torch
diff --git a/pyproject.toml b/pyproject.toml
index 5c13db0c..98dd91d7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,9 +1,9 @@
 [tool.poetry]
 name = "assume-framework"
-version = "0.2.0"
+version = "0.2.1"
 description = "ASSUME - Agent-Based Electricity Markets Simulation Toolbox"
 authors = ["ASSUME Developers <contact@assume-project.de>"]
-license = "LICENSE"
+license = "AGPL-3.0-or-later"
 readme = "README.md"
 
 homepage = "https://assume.readthedocs.io"
@@ -12,9 +12,8 @@ repository = "https://github.com/assume-framework/assume"
 keywords = ["agent based simulation", "energy market", "reinforcement learning", "market simulation", "simulation"]
 
 classifiers=[
-    "Development Status :: 3 - Alpha",
+    "Development Status :: 4 - Beta",
     "Intended Audience :: Developers",
-    "License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)",
     "Operating System :: OS Independent",
     "Framework :: AsyncIO",
     "Topic :: Scientific/Engineering",