Update how state works in UI