Dodal delete movies

controllers/movieController.js

@@ -104,16 +104,21 @@ module.exports = {
      */
     remove: function (req, res) {
         var id = req.params.id;
-
-        MovieModel.findByIdAndRemove(id, function (err, movie) {
-            if (err) {
-                return res.status(500).json({
-                    message: 'Error when deleting the movie.',
-                    error: err
+        console.log(id);
+        MovieModel.findOneAndDelete({ code: id })
+        .then(movie => {
+            if (!movie) {
+                return res.status(404).json({
+                    message: 'No such movie'
                 });
             }
-
             return res.status(204).json();
+        })
+        .catch(err => {
+            return res.status(500).json({
+                message: 'Error when deleting the movie.',
+                error: err
+            });
         });
     },
 };