Add imports for Foundation and UIKit