import UIKit (fixes compiling problem) in extensions