diff --git a/CMakeLists.txt b/CMakeLists.txt
index d0cfb7a..8dd06e2 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -1,5 +1,6 @@
-set(COMPONENT_ADD_INCLUDEDIRS .)
+# The following lines of boilerplate have to be in your project's CMakeLists
+# in this exact order for cmake to work correctly
+cmake_minimum_required(VERSION 3.5)
 
-set(COMPONENT_SRCS "ssd1306.c")
-
-register_component()
+include($ENV{IDF_PATH}/tools/cmake/project.cmake)
+project(ssd1306)
diff --git a/main/CMakeLists.txt b/main/CMakeLists.txt
new file mode 100644
index 0000000..c931a02
--- /dev/null
+++ b/main/CMakeLists.txt
@@ -0,0 +1,4 @@
+set(COMPONENT_SRCS main.c ssd1306.c ssd1306_i2c.c ssd1306_spi.c)
+set(COMPONENT_ADD_INCLUDEDIRS ".")
+
+register_component()