     /// - Returns: UIView
     open func pickerView(_ pickerView: UIPickerView, viewForRow row: Int, forComponent component: Int, reusing view: UIView?) -> UIView {
         var resultView: CountryView
-        
+
         if view == nil {
             resultView = CountryView()
         } else {
             resultView = view as! CountryView
         }
-        
+
         resultView.setup(countries[row])
         if !showPhoneNumbers {
             resultView.countryCodeLabel.isHidden = true
         }
         return resultView
     }
-    
+
     /// Function for handing data from UIPickerView
     ///
     /// - Parameters:

        self.phoneCode = phoneCode
        self.flagName = flagName
    }
    
    var flag: UIImage? {
        return UIImage(named: flagName, in: Bundle(for: CountryPicker.self), compatibleWith: nil)
    }
}

open class CountryPicker: UIPickerView, UIPickerViewDelegate, UIPickerViewDataSource {
    
    var countries: [Country]!
    open weak var countryPickerDelegate: CountryPickerDelegate?
    open var showPhoneNumbers: Bool = true
    
    /// init
    ///
    /// - Parameter frame: initialization

        super.init(frame: frame)
        setup()
    }

    required public init?(coder aDecoder: NSCoder) {
        super.init(coder: aDecoder)
        setup()
    }

    /// Setup country code picker
    func setup() {
        countries = CountryPicker.countryNamesByCode()

        super.dataSource = self
        super.delegate = self
    }

    // MARK: - Country Methods

    /// setCountry
    ///
    /// - Parameter code: selected country
