app/static/js/word_operation.js

@@ -61,6 +61,12 @@ function delete_word(theWord) {
             } else {
                 $("#p_" + theWord).remove();
             }
+	    // remove highlighting for the word
+	    let highlightedWords = document.querySelectorAll('.highlighted');
+	    for (let x of highlightedWords) {
+		if (x.innerHTML == word)
+		    x.replaceWith(x.innerHTML);
+	    }
         }
     });
 }