Removed UIKit class restriction