Updated UIViewController category mechanism