Added support for Objective-C categories