diff --git a/pomodoroprompt.py b/pomodoroprompt.py
index 0b4cfae..becd772 100644
--- a/pomodoroprompt.py
+++ b/pomodoroprompt.py
@@ -1,11 +1,12 @@
 import sys
 import csv
+# This is a local script are we supposed to indicate that?
+import prompt_window
 import time
 from datetime import datetime, timedelta
 from pathlib import Path
 from pytz import timezone
 from playsound import playsound
-from prompt_window import PromptWindow
 
 import pytz
 
@@ -27,8 +28,7 @@ def pomodoro_prompt_plan(whatdid = ''):
     # than these pop-up GUI notifications/text entry.
     if whatdid:
         text = "\n\n(Last thing you did: " + whatdid + ')\n'
-    prompt = PromptWindow(prompt=text, task=whatdid)
-    whatnext = prompt.retrieve()
+    whatnext = prompt_window.prompt(prompt=text, task=whatdid)
     # Pressing cancel returns None, but we want to just treat it as an empty string.
     if whatnext is None:
         whatnext = ''