Changed @import to #import for UIKit