Update ViewController.m

made changes to work with simplified code.