Как автоматически изменить размер UITableViewCell, когда я хочу добавить динамический текст

Я настраиваю UITableView с несколькими ячейками. Когда я удерживаю ячейку, появляется меню, и я могу добавить комментарий/текст в свою ячейку. Я хочу, чтобы не отображалось пустое пространство меток комментариев, если пользователь не вставляет комментарий. Но если пользователь вставит комментарий, тогда появится этот ярлык и отобразится весь вставленный комментарий. Возможно ли это сделать без жесткого кода высоты ячейки? Потому что это приложение будет работать на нескольких устройствах, и экран на каждом из них будет разным.

Я уже безуспешно пытался использовать авто-макет, используя эти строки кода в viewDidLoad:

questionsTableView.rowHeight = UITableView.automaticDimension
questionsTableView.estimatedRowHeight = 390

Также я пытался использовать questionsTableView.beginUpdates до добавления комментария к метке и questionsTableView.endUpdates после того, как комментарий был назначен метке, но тоже не работает.

Мне не удается отобразить ячейку таким образом, когда пользователь запускает приложение, и если пользователь добавляет какой-либо комментарий, затем расширяет ячейку и отображает метку:

Вот мой код:

class MainVC: UIViewController {

    // Interface Links
    @IBOutlet weak var questionsTableView: UITableView!

    // Properties
    var imagePicker: UIImagePickerController!
    var firstCommentReceived = ""
    var secondCommentReceived = ""
    var images = [UIImage]()

    override func viewDidLoad() {
        super.viewDidLoad()

        questionsTableView.rowHeight = UITableView.automaticDimension
        questionsTableView.estimatedRowHeight = 390
    }
}

extension MainVC: UITableViewDelegate, UITableViewDataSource {

    func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {

        return 2
    }

    func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {

        let cell = tableView.dequeueReusableCell(withIdentifier: "customCell", for: indexPath) as! MainCell
        cell.delegate = self
        cell.configCell()
        if !firstCommentReceived.isEmpty{
            cell.firstCommentLabel.text = firstCommentReceived
        }
        if !secondCommentReceived.isEmpty{
            cell.secondCommentLabel.text = secondCommentReceived
        }
        if !images.isEmpty{
            cell.selectedImageView.image = images[0]
        }
        return cell
    }

    // Set the height of each row from UITableView
    func tableView(_ tableView: UITableView, heightForRowAt indexPath: IndexPath) -> CGFloat {

        return 390 // size of the maximum height of the cell if the user add comments and images. This size should be dynamic not hardcoded.
    }
}

Здесь я создал небольшой пример GitHub с моей проблемой: https://github.com/tygruletz/sendDataFromCellToController

Спасибо, если вы это читаете!