diff --git a/setup.py b/setup.py
index 7fe0131..e6cb4c8 100644
--- a/setup.py
+++ b/setup.py
@@ -1,6 +1,11 @@
+import glob
 from setuptools import setup
 from pybind11.setup_helpers import Pybind11Extension, build_ext
 
+sources = ['binder.cpp']
+sources += glob.glob("tinytransfer/*.cpp")
+
+
 # Convert distutils Windows platform specifiers to CMake -A arguments
 PLAT_TO_CMAKE = {
     "win32": "Win32",
@@ -9,10 +14,22 @@
     "win-arm64": "ARM64",
 }
 
+# explicitly added heatshrink paths
+sources += [
+    "tinytransfer/heatshrink/heatshrink.c",
+    "tinytransfer/heatshrink/heatshrink_decoder.c",
+    "tinytransfer/heatshrink/heatshrink_encoder.c",
+]
+
 ext_modules = [
     Pybind11Extension(
         "pytinytransfer",
-        ['binder.cpp'],
+        sources,
+        #tinytransfer and heatshrink headers
+        include_dirs=[
+            "tinytransfer",
+            "tinytransfer/heatshrink",
+        ],
     ),
 ]