diff --git a/prompt_window.py b/prompt_window.py
index f097fb4..f5b8275 100644
--- a/prompt_window.py
+++ b/prompt_window.py
@@ -55,11 +55,18 @@ class PromptWindow(Gtk.Window):
         self.entry.set_text(self.orig_task)
 
     def retrieve(self):
-        return self.task
+        if self.task:
+            return self.task
+        else:
+            return None
+        # started to do conditional like this but maybe main program handles.
+        # else if self.orig_task:
+        #    return "Pomodoro cancelled, original goal was:"
 
 
-win = PromptWindow()
-win.show_all()
-Gtk.main()
-tha_task = win.retrieve()
-print(tha_task)
+if __name__ == '__main__':
+    win = PromptWindow("I'mma tell you what i'm gonna do.")
+    win.show_all()
+    Gtk.main()
+    tha_task = win.retrieve()
+    print(tha_task)