Cleanup config flow (#2932)

* Break up config flow dialog

* Allow picking devices when config flow finishes

* Lint

* Tweaks

src/components/entity/state-badge.ts

@@ -39,7 +39,7 @@ class StateBadge extends LitElement {
   }
 
   protected updated(changedProps: PropertyValues) {
-    if (!changedProps.has("stateObj")) {
+    if (!changedProps.has("stateObj") || !this.stateObj) {
       return;
     }
     const stateObj = this.stateObj;