CountryTableViewCell.swift

import UIKit

class CountryTableViewCell: UITableViewCell {
    
    // MARK: - Initialization
    override init(style: UITableViewCell.CellStyle, reuseIdentifier: String?) {
        super.init(style: style, reuseIdentifier: reuseIdentifier)
        setupUI()
    }
    
    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    // MARK: - Properties
    var viewModel: CountryViewModel! {
        didSet {
            self.configure()
        }
    }
    
    .....
 
}

// MARK: - Configuration
extension CountryTableViewCell {
    func configure() {
        countryNameLabel.text = viewModel.name
    }
}
.....