Changed from ViewController to UIViewController