Update ViewController.h

simplified code.