Now a category on NSObject instead of UIViewController