Tidy up
Add UIKit imports