diff --git a/notification.go b/ui/notification.go
similarity index 96%
rename from notification.go
rename to ui/notification.go
index 584434a..4741d34 100644
--- a/notification.go
+++ b/ui/notification.go
@@ -1,4 +1,4 @@
-package main
+package ui
 
 import (
 	"time"
@@ -9,6 +9,14 @@ import (
 	"github.com/gdamore/tcell/v2"
 )
 
+var (
+	UI *Application
+)
+
+func ConnectUI(a *Application) {
+	UI = a
+}
+
 /* Notification Primitive */
 type Notification struct {
 	*tview.Box