diff --git a/setup.py b/setup.py
index f8cf4a3..bb472eb 100644
--- a/setup.py
+++ b/setup.py
@@ -1,4 +1,5 @@
 from setuptools import setup
+import os
 
 setup(
    name='prtools',
@@ -8,4 +9,7 @@
    author_email='',
    packages=['prtools'],
    install_requires=['sklearn', 'numpy', 'matplotlib', 'requests', 'mlxtend'],
+   package_data={
+      'prtools': [ os.path.join('data', '*.mat') ],
+   },
 )