diff --git a/gravity.php b/gravity.php
index ee606d56..3cb41228 100644
--- a/gravity.php
+++ b/gravity.php
@@ -16,7 +16,8 @@
     Success!
 </div>
 
-<pre id="output" style="width: 100%; height: 100%;"></pre>
+<button class="btn btn-lg btn-primary btn-block" id="gravityBtn">Update Lists</button>
+<pre id="output" style="width: 100%; height: 100%;" hidden="true"></pre>
 
 <?php
     require "footer.php";
diff --git a/js/pihole/gravity.js b/js/pihole/gravity.js
index 04691c1c..cd9bac3a 100644
--- a/js/pihole/gravity.js
+++ b/js/pihole/gravity.js
@@ -1,9 +1,11 @@
-function eventsourcetest() {
+function eventsource() {
     var alInfo = $("#alInfo");
     var alSuccess = $("#alSuccess");
-    var ta = document.getElementById("output");
+    var ta = $("#output");
     var source = new EventSource("php/gravity.sh.php");
 
+    ta.html("");
+    ta.show();
     alInfo.show();
     alSuccess.hide();
 
@@ -13,7 +15,7 @@ function eventsourcetest() {
             alSuccess.show();
         }
 
-        ta.innerHTML += e.data;
+        ta.append(e.data);
 
     }, false);
 
@@ -21,11 +23,13 @@ function eventsourcetest() {
     source.addEventListener("error", function(e) {
         alInfo.delay(1000).fadeOut(2000, function() { alInfo.hide(); });
         source.close();
+        $("#gravityBtn").removeAttr("disabled");
     }, false);
 }
 
-$(function(){
-   eventsourcetest();
+$("#gravityBtn").on("click", () => {
+    $("#gravityBtn").attr("disabled", true);
+    eventsource();
 });
 
 // Handle hiding of alerts