diff --git a/dist/squeeze_and_excitation-1.0-py2.py3-none-any.whl b/dist/squeeze_and_excitation-1.0-py2.py3-none-any.whl
index 585e5d4..212ec1c 100644
Binary files a/dist/squeeze_and_excitation-1.0-py2.py3-none-any.whl and b/dist/squeeze_and_excitation-1.0-py2.py3-none-any.whl differ
diff --git a/setup.py b/setup.py
index 76bf219..44aaea0 100644
--- a/setup.py
+++ b/setup.py
@@ -7,4 +7,5 @@
                  author_email="shayan.siddiqui89@gmail.com",
                  description="Squeeze and Excitation pytorch implementation",
                  packages=setuptools.find_packages(),
-                 install_requires=['torch'])
+                 install_requires=['numpy==1.14.3','torch==1.0.1.post2'],
+                 python_requires='>=3.5')