diff --git a/solution.js b/solution.js
index 183e8f6429098340fd54e4d15f3eab8d55d9bbae..c06a950e5bf955c99a082acab806fac40f45c44d 100644
--- a/solution.js
+++ b/solution.js
@@ -67,6 +67,10 @@ module.exports.removeMany = ids => {
     return Solution.remove({_id: {$in: ids}});
 };
 
+module.exports.update = (id, update) => {
+    return Solution.findOneAndUpdate({_id: id}, update);
+};
+
 module.exports.save = solution => {
     return solution.save()
 };
@@ -158,3 +162,7 @@ module.exports.setStatus = (id, status) => {
 module.exports.setTestingResult = (id, testingResult) => {
     return Solution.findOneAndUpdate({_id: id}, {testingResult});
 };
+
+module.exports.setPlagiarism = (id, plagiarism) => {
+    return Solution.findOneAndUpdate({_id: id}, {plagiarism});
+};