Handle UDAs for editing; Fix layout; Add annotations

common/common.go

@@ -15,7 +15,7 @@ type Common struct {
 	TW     taskwarrior.TaskWarrior
 	Keymap *Keymap
 	Styles *Styles
-	Udas   []string
+	Udas   []taskwarrior.Uda
 
 	pageStack *Stack[Component]
 	width     int
@@ -54,5 +54,11 @@ func (c *Common) PushPage(page Component) {
 }
 
 func (c *Common) PopPage() (Component, error) {
-	return c.pageStack.Pop()
+	component, err := c.pageStack.Pop()
+	if err != nil {
+		return nil, err
+	}
+
+	component.SetSize(c.width, c.height)
+	return component, nil
 }