override func viewWillAppear(_ animated: Bool) {
    super.viewWillAppear(true)

    if let indexPath = savedIndexPath {
        messageArray[indexPath.row].flag = true
        messagesTable.reloadRows(at: [indexPath], with: .none)
    }
}

func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
    // Save indexPath to remember scroll position
    self.savedIndexPath = indexPath
}