edit demo for viewmodel style