reworking viewModel to include input and checkbox