diff --git a/CMakeLists.txt b/CMakeLists.txt
index e686547..1dad756 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -6,6 +6,11 @@ set(PYBIND11_CPP_STANDARD -std=c++17)
 
 add_subdirectory("lib/pybind11")
 
-pybind11_add_module(NDIlib SHARED ${CMAKE_SOURCE_DIR}/src/main.cpp)
+file(GLOB INCLUDES ${CMAKE_CURRENT_SOURCE_DIR}/src/*.hpp)
+file(GLOB SOURCES ${CMAKE_CURRENT_SOURCE_DIR}/src/*.cpp)
+
+pybind11_add_module(NDIlib SHARED ${INCLUDES} ${SOURCES})
 
 target_link_libraries(NDIlib PRIVATE pybind11::module ndi)
+
+source_group("" FILES ${SOURCES} ${INCLUDES})