Remove unnecessary UIKit imports