From e9a6ffd2b80e1a0098ce17f9c9713727780b8875 Mon Sep 17 00:00:00 2001
From: Dan Jones <danjones@goodevilgenius.org>
Date: Tue, 1 Sep 2020 11:35:45 -0500
Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=A8=20pick-show:=20Add=20refreshPage?=
 =?UTF-8?q?=20function=20to=20encapsulate=20page=20refresh?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 trakt.pick-show.user.js | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/trakt.pick-show.user.js b/trakt.pick-show.user.js
index 28005b6..b425424 100644
--- a/trakt.pick-show.user.js
+++ b/trakt.pick-show.user.js
@@ -16,6 +16,8 @@
 (function() {
     'use strict';
 
+    let watched_shows;
+
     $(window).konami({
         code: [80, 73, 67, 75],
         cheat: pickShow
@@ -32,14 +34,17 @@
             $leftNav.append('<span class="filter-dropdown toggle-simple-progress pick-episode" title="Pick Episode"><span class="icon trakt-icon-wand"></span></span>')
                 .find('.pick-episode').on('click', pickShow);
         }
-
-        $('div[data-type="show"]').on('click', unghost);
     }
-    addPickButton();
-    const observer = new MutationObserver(addPickButton);
-    observer.observe(document.head.parentElement, {childList: true});
 
-    let watched_shows;
+    function refreshPage() {
+        addPickButton();
+        $('div[data-type="show"]')
+            .on('click', unghost);
+    }
+
+    refreshPage();
+    const observer = new MutationObserver(refreshPage);
+    observer.observe(document.head.parentElement, {childList: true});
 
     function getWeightedIndex(total) {
         const opts = [];