diff --git a/src/markdownSyntaxHighlight.js b/src/markdownSyntaxHighlight.js
index d909d7c..57cc68e 100644
--- a/src/markdownSyntaxHighlight.js
+++ b/src/markdownSyntaxHighlight.js
@@ -28,13 +28,16 @@ module.exports = function(str, language) {
let lines = html.split("\n").slice(0, -1); // The last line is empty.
let highlightedLines = lines.map(function(line, j) {
- return "" +
- line +
- "";
+ if (highlights.isHighlighted(j)) {
+ return "" + line + "";
+ }
+ if (highlights.isHighlightedAdd(j)) {
+ return "" + line + "";
+ }
+ if (highlights.isHighlightedRemove(j)) {
+ return "" + line + "";
+ }
+ return "" + line + "";
});
return `
${highlightedLines.join("
")}
`;