diff --git a/main.cpp b/main.cpp
index 6faba55..f4313f0 100644
--- a/main.cpp
+++ b/main.cpp
@@ -1,3 +1,5 @@
+#define  STB_IMAGE_WRITE_IMPLEMENTATION
+#include <stb_image_write.h>
 #include "rainbow.h"
 #include "mandel.h"
 
diff --git a/stbiw/CMakeLists.txt b/stbiw/CMakeLists.txt
index b56b853..9b17f0a 100644
--- a/stbiw/CMakeLists.txt
+++ b/stbiw/CMakeLists.txt
@@ -1 +1,2 @@
-message(FATAL_ERROR "请修改 stbiw/CMakeLists.txt！要求生成一个名为 stbiw 的库")
+add_library(stbiw INTERFACE)
+target_include_directories(stbiw INTERFACE .)