Revert "UiState stored in implementation."

This reverts commit 9b9a52d5491159e9ab09bd9c00c2106cc96793b0.
Revert "Revert "UiState stored in implementation.""

This reverts commit f10e452a48ddbd22a3407378dedea1428bd2bd50.