diff --git a/README.md b/README.md
index 0c4e2e72..a42f84eb 100644
--- a/README.md
+++ b/README.md
@@ -13,7 +13,7 @@ Context-specific parameters are essential for:
 
 ## Install and Use Contextualized
 ```
-pip install git+https://github.com/cnellington/Contextualized.git
+pip install contextualized-ml
 ```
 
 Take a look at the [main demo](docs/demos/main_demo.ipynb) for a complete overview with code, or the [easy demo](docs/demos/Easy-demo/easy_demo.ipynb) for a quickstart with sklearn-style wrappers!
diff --git a/docs/installation.ipynb b/docs/installation.ipynb
index b06ad8f5..0f1ad93d 100644
--- a/docs/installation.ipynb
+++ b/docs/installation.ipynb
@@ -12,7 +12,7 @@
     "```{tab-item} pip\n",
     "\n",
     "`\n",
-    "pip install https://github.com/cnellington/contextualized.git\n",
+    "pip install contextualized-ml\n",
     "`\n",
     "\n",
     "```\n",
@@ -27,6 +27,14 @@
     "\n",
     "Now, let's walk through an example of Contextualized analysis."
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "9dce20f5",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {
diff --git a/setup.py b/setup.py
index b7b7c48d..afd6f149 100644
--- a/setup.py
+++ b/setup.py
@@ -1,15 +1,22 @@
-import setuptools
+"""
+Setup and requirements for Contextualized.ML
+"""
 
-setuptools.setup(name='contextualized',
-    packages=[
-        'contextualized',
-        'contextualized.regression',
-        'contextualized.dags',
-        'contextualized.easy',
-    ],
-    version='0.2.1',
+from setuptools import find_packages, setup
+
+DESCRIPTION = "An ML toolbox for estimating context-specific parameters."
+VERSION = '0.2.1.2'
+
+setup(
+    name='contextualized',
+    author="Contextualized.ML team",
+    version=VERSION,
+    description=DESCRIPTION,
+    url="https://github.com/cnellington/contextualized",
+    packages=find_packages(),
     install_requires=[
         'lightning',
+        'pytorch-lightning',
         'torch',
         'torchvision',
         'numpy',