Import UIKit instead of Foundation in places where needed