diff --git a/CMakeLists.txt b/CMakeLists.txt
index 64ed854..46079cf 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -10,11 +10,18 @@ find_package(SDL2_image REQUIRED)
 find_package(SDL2_ttf REQUIRED)
 
 add_executable(Obstacle
-        Main.cpp
-        Game.cpp
-        Game.h
-        Helper.cpp
-        Helper.h
+        src/main.cpp
+        src/Renderer.cpp
+        src/Renderer.h
+        src/states/GameState.h
+        src/states/InGameState.cpp
+        src/states/InGameState.h
+        src/states/MainMenuState.cpp
+        src/states/MainMenuState.h
+        src/entities/Player.cpp
+        src/entities/Player.h
+        src/entities/Opponent.cpp
+        src/entities/Opponent.h
 )
 
 target_link_libraries(Obstacle SDL2::SDL2 SDL2_image::SDL2_image SDL2_ttf::SDL2_ttf)
\ No newline at end of file