Modify viewController.m to accomodate changes in API