diff --git a/.gitignore b/.gitignore
index 2d97991d7e..f098a90aab 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,5 @@
 .DS_Store
 __pycache__/
+build/
+*.egg-info/
 
diff --git a/setup.py b/setup.py
index 461626e1e9..8e46902553 100644
--- a/setup.py
+++ b/setup.py
@@ -1,7 +1,9 @@
-from setuptools import setup
+from setuptools import setup, find_packages
 
 setup(
     name="improved-diffusion",
-    py_modules=["improved_diffusion"],
-    install_requires=["blobfile>=1.0.5", "torch", "tqdm"],
+    packages = find_packages(),
+    install_requires=["blobfile>=1.0.5", "torch", "tqdm", "mpi4py"],
+    license=('LICENSE'),
+    version='0.0.1'
 )