Update CMakeLists.txt and main.cpp