diff --git a/setup.py b/setup.py
index f766f03..c1d922a 100644
--- a/setup.py
+++ b/setup.py
@@ -9,9 +9,9 @@
 setup(
     name='pyglom',
     packages=find_packages(),
-    version='0.0.1',
+    version='0.0.2',
     license='MIT',
-    description='Pytorch implementation of GLOM',
+    description='PyTorch implementation of GLOM. PyTorch-Lightning version is also available.',
     author='Yeonwoo Sung',
     author_email='neos960518@gmail.com',
     url='https://github.com/YeonwooSung/GLOM',
@@ -28,6 +28,7 @@
     ],
     install_requires=[
         'einops>=0.3',
+        "pytorch-lightning==0.8.5",
         'torch>=1.6'
     ],
     classifiers=[