Add support for UIView instead of UIImage.