Import UIKit/Cocoa headers to fix compile error