Add #import <UIKit/UIKit.h>