Added missing UIKit import