Changed to use UIKit for color