Made proper use of a ViewModel rather than the ViewBag