Using ViewModel correctly. Hopefully.