Adding UIKit to make things work