Update ViewController.m

Hope this helps