diff --git a/gifit/__init__.py b/gifit/__init__.py
index e3b72c3..09462af 100644
--- a/gifit/__init__.py
+++ b/gifit/__init__.py
@@ -1,4 +1,4 @@
-__version__ = '1.0.0'
+__version__ = '0.1.0'
 
 DISPLAY_TITLE = r"""
        _        _                                  __ _ _   
diff --git a/setup.py b/setup.py
index 2f48a2d..aea340a 100644
--- a/setup.py
+++ b/setup.py
@@ -53,7 +53,7 @@ def get_version(rel_path: str) -> str:
         ]
     },
     package_data={
-        'gifit': ['models']
+        'gifit': ['models/*']
     },
     scripts=scripts
 )