Update ViewController.m

Simplified Code