diff --git a/setup.py b/setup.py
index d694061..25414f8 100644
--- a/setup.py
+++ b/setup.py
@@ -1,5 +1,4 @@
 from setuptools import setup
-from setuptools import Extension
 
 with open('requirements/requirements.txt') as f:
     requirements = f.readlines()
@@ -11,9 +10,12 @@
     description='GANs2.0',
     long_description=(
         'Generative Adversarial Networks in TensorFlow 2.0.'
-        ),
+    ),
     long_description_content_type='text/markdown',
     install_requires=requirements,
+    extras_require={
+        'tensorflow': ['tensorflow>=2.2.0'],
+    },
     packages=[
         'gans.datasets',
         'gans.layers',
@@ -22,4 +24,4 @@
     ],
     zip_safe=True
 )
-'Generative Adversarial Networks in TensorFlow 2.0'
\ No newline at end of file
+'Generative Adversarial Networks in TensorFlow 2.0'