1 files changed, 7 insertions, 4 deletions

diff --git a/uncompressed/gmail/webview.js b/uncompressed/gmail/webview.js
index 521d4ad..febf64b 100644
--- a/uncompressed/gmail/webview.js
+++ b/uncompressed/gmail/webview.js
@@ -1,6 +1,6 @@
-"use strict";
+import path from 'path';
 
-module.exports = Franz => {
+module.exports = (Franz) => {
   const getMessages = function getMessages() {
     let count = 0;
 
@@ -10,14 +10,17 @@ module.exports = Franz => {
       }
     }
 
+    // Just incase we don't end up with a number, set it back to zero (parseInt can return NaN)
     count = parseInt(count, 10);
-
     if (isNaN(count)) {
       count = 0;
     }
 
+    // set Franz badge
     Franz.setBadge(count);
   };
 
+  Franz.injectCSS(path.join(__dirname, 'service.css'));
+  // check for new messages every second and update Franz badge
   Franz.loop(getMessages);
-};
\ No newline at end of file
+};