Update ViewController.h

Simplified Code