diff --git a/setup.py b/setup.py
index 4f2e3b980052f1d1d32e66b14e1d203d0249afcf..2ca34f3421353419a7841ae4b1a495dfbfacd7a8 100644
--- a/setup.py
+++ b/setup.py
@@ -1,19 +1,22 @@
-from setuptools import setup
+from setuptools import setup, find_packages
 
 setup(
     name='neural_filters',
-    version='1.0',
+    version='0.2',
     description='Linear filters for neural networks in pyTorch',
     author='Idiap research institute - Francois Marelli',
     author_email='francois.marelli@idiap.ch',
+    license='GNU GPL v3',
     classifiers=[
         'Development Status :: 5 - Production/Stable',
         'Intended Audience :: Developers',
         'License :: OSI Approved :: GNU GPL v3',
         'Programming Language :: Python :: 3.6',
     ],
+    packages=find_packages('neural_filters'),
     install_requires=[
         'torch>=0.4.0',
         'numpy',
         ],
+    zip_safe=True,
 )