Add: missing UI elements