Import UIKit for UIImage