diff --git a/FlippR-Driver/build/CMakeLists.txt b/FlippR-Driver/build/CMakeLists.txt
index 0308809..c4646a2 100644
--- a/FlippR-Driver/build/CMakeLists.txt
+++ b/FlippR-Driver/build/CMakeLists.txt
@@ -1,26 +1,25 @@
 cmake_minimum_required(VERSION 3.9.1)
-        project(flippr-driver)
+project(flippr-driver)
 
 set(CMAKE_BUILD_TYPE Debug)
 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")  # enable C++11 standard
 
-set(CMAKE_SOURCE_DIR ../src)
-
 set(CMAKE_BINARY_DIR ${CMAKE_SOURCE_DIR}/bin)
 set(EXECUTABLE_OUTPUT_PATH ${CMAKE_BINARY_DIR})
 set(LIBRARY_OUTPUT_PATH ${CMAKE_BINARY_DIR})
 set(SOURCE_DIR ../src)
 
 find_package(Threads)
+find_package(Boost)
 
 include_directories(${SOURCE_DIR}/input)
 include_directories(${SOURCE_DIR}/output)
 include_directories(${SOURCE_DIR}/lib)
 include_directories(${SOURCE_DIR}/utilities)
 
-file(GLOB SOURCES "src/*/*.cpp")
+file(GLOB SOURCES ${SOURCE_DIR}/*/*.cpp")
+file(GLOB HEADER_SOURCES ${SOURCE_DIR}/*/*.hpp")
 
-
-add_library(flippr_driver STATIC ${SOURCES})
+add_library(flippr_driver STATIC ${SOURCES} ${HEADER_SOURCES})
     
 target_link_libraries(flippr_driver ${CMAKE_SOURCE_DIR}/lib/libwiringPi.so.2.44)