Fixed missing UIKit import

Importing UIKit is required to use UIColor.