Import UIKit instead of Foundation