diff --git a/js/app.js b/js/app.js
index bbf9973..2dd136b 100644
--- a/js/app.js
+++ b/js/app.js
@@ -10,6 +10,6 @@ jQuery.noConflict();
 (($) => {
   // When DOM is ready
   $(() => {
-    dotsEffect.init();
+    dotsEffect($);
   });
 })(jQuery);
diff --git a/js/modules/dots.js b/js/modules/dots.js
index 7c9ee77..69fcd92 100644
--- a/js/modules/dots.js
+++ b/js/modules/dots.js
@@ -2,31 +2,25 @@
 //
 // You can remove or add your own function in this file.
 
-const dotsEffect = (($) => {
-  const createCells = () => {
-    const width = 10;
-    const height = 10;
-    const size = width * height;
-    let html = '<div class="entry-cells">';
+const createCells = () => {
+  const width = 10;
+  const height = 10;
+  const size = width * height;
+  let html = '<div class="entry-cells">';
 
-    for (let i = 0; i < size; i += 1) {
-      html += `<div class="cell cell-${i}"></div>`;
-    }
+  for (let i = 0; i < size; i += 1) {
+    html += `<div class="cell cell-${i}"></div>`;
+  }
 
-    html += '</div>';
+  html += '</div>';
 
-    return html;
-  };
+  return html;
+};
 
-  const init = () => {
-    const $cnt = $('.entry-section');
-    const cells = createCells();
-    $cnt.html(cells);
-  };
+const init = ($) => {
+  const $cnt = $('.entry-section');
+  const cells = createCells();
+  $cnt.html(cells);
+};
 
-  return {
-    init,
-  };
-})(jQuery);
-
-export default dotsEffect;
+export default init;