diff --git a/lua/cp/constants.lua b/lua/cp/constants.lua
index acabc3c..13cb27d 100644
--- a/lua/cp/constants.lua
+++ b/lua/cp/constants.lua
@@ -8,11 +8,8 @@ M.PYTHON = 'python'
 
 ---@type table<string, string>
 M.filetype_to_language = {
-  cc = M.CPP,
-  cxx = M.CPP,
+  python = M.PYTHON,
   cpp = M.CPP,
-  py = M.PYTHON,
-  py3 = M.PYTHON,
 }
 
 ---@type table<string, string>
diff --git a/lua/cp/snippets.lua b/lua/cp/snippets.lua
index e0237c8..ac43a65 100644
--- a/lua/cp/snippets.lua
+++ b/lua/cp/snippets.lua
@@ -96,7 +96,11 @@ if __name__ == "__main__":
 if __name__ == "__main__":
     solve()]],
 
-      cses = [[{}]],
+      cses = [[def solve():
+    {}
+
+if __name__ == "__main__":
+    solve()]],
     },
   }