diff --git a/assignments/favourite_movie/favourite_movie.js b/assignments/favourite_movie/favourite_movie.js
index a8148eb..64ffcf3 100644
--- a/assignments/favourite_movie/favourite_movie.js
+++ b/assignments/favourite_movie/favourite_movie.js
@@ -2,6 +2,13 @@ const movies = [];
 
 function favouriteMovie(operation, movie) {
     // Write your code here
+    if (operation==="add"){
+        movies.push(movie);
+    }
+    else if (operation==="remove"){
+        movies.pop(movie);
+    }
+    return movies
 }
 
 module.exports = favouriteMovie;