diff --git a/apiv1/controllers/CourseController.js b/apiv1/controllers/CourseController.js
index 063c27a5..fbe480df 100644
--- a/apiv1/controllers/CourseController.js
+++ b/apiv1/controllers/CourseController.js
@@ -192,6 +192,7 @@ module.exports = {
                     Course.shortname = req.body.shortname ? req.body.shortname : Course.shortname;
                     Course.lessons = req.body.lessons ? req.body.lessons : Course.lessons;
                     Course.difficulty = req.body.difficulty ? req.body.difficulty : Course.difficulty;
+                    Course.categories = req.body.categories ? req.body.categories : Course.categories;
                     Course.description = req.body.description ? req.body.description : Course.description;
                     Course.lessons = req.body.lessons ? req.body.lessons : Course.lessons;
 
@@ -235,4 +236,4 @@ module.exports = {
         });
 
     },
-};
\ No newline at end of file
+};
diff --git a/apiv1/models/CourseModel.js b/apiv1/models/CourseModel.js
index 9978a42a..145e176c 100644
--- a/apiv1/models/CourseModel.js
+++ b/apiv1/models/CourseModel.js
@@ -12,9 +12,16 @@ let LessonSchema = new Schema({
 let CourseSchema = new Schema({
     'shortname': String,
     'name': String,
-    'difficulty': Number,
+    'difficulty': {
+        type: Number,
+        enum: [0, 1, 2, 3],
+    },
+    'categories': [{
+        type: String,
+        enum: ["geometry", "transformations", "animations", "groups", "firstTimer", "teachers", "misc"],
+    }],
     'description': String,
     'lessons': [LessonSchema]
 });
 
-module.exports = mongoose.model('Course', CourseSchema);
\ No newline at end of file
+module.exports = mongoose.model('Course', CourseSchema);