diff --git a/prompt_window.py b/prompt_window.py
index de86a1f..17ce26c 100644
--- a/prompt_window.py
+++ b/prompt_window.py
@@ -6,7 +6,7 @@ from gi.repository import Gtk, GLib
 
 class PromptWindow(Gtk.Window):
 
-    def __init__(self, task=None):
+    def __init__(self, prompt=None, task=None):
 
         if task is None:
             task = ""
@@ -41,7 +41,8 @@ class PromptWindow(Gtk.Window):
 
         vbox.pack_start(hbox, True, True, 0)
 
-        # self.set_title("Entry")
+        if prompt:
+            self.set_title(prompt)
 
         self.show_all()
         Gtk.main()