diff --git a/CMakeLists.txt b/CMakeLists.txt
index d6bd42a9353874df93109f444e0c281cc859a99a..a6401c510279b036255b9befed570129e031b3e9 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -6,9 +6,13 @@ set(CMAKE_CXX_STANDARD 11)
 set(CMAKE_CXX_FLAGS "-O3 -Wall")
 
 set(HEADERS lib/nBodySim.hpp
+            lib/node.hpp
+            lib/tree.hpp
             )
 
 set(SOURCES lib/nBodySim.cpp
+            lib/node.cpp
+            lib/tree.cpp
             )
 
 add_library(nbodyLib ${HEADERS} ${SOURCES})