diff --git a/prisma/schema.prisma b/prisma/schema.prisma
index 9b3ad08..55a417b 100644
--- a/prisma/schema.prisma
+++ b/prisma/schema.prisma
@@ -36,4 +36,18 @@ model Problem {
   published   Boolean    @default(false)
   authorId    Int
   author      User       @relation(fields: [authorId], references: [id])
+  Template    Template[]
+}
+
+enum EditorLanguage {
+  C
+  CPP
+}
+
+model Template {
+  id        Int            @id @default(autoincrement())
+  language  EditorLanguage
+  template  String
+  problemId Int
+  problem   Problem        @relation(fields: [problemId], references: [id])
 }