Corrected import statement - import UIKit.h instead of Foundation, as we are using UI-related classes