Make sure to @import UIKit