Added UIKit import to help swift compatibility