Updated Edit view to use ViewModel.