Jetpack compose - MVVM State management in a simple way