Update ViewController.m

Edit from github