 
 class NoteEditViewController : AppViewController, UITextViewDelegate {
     
-    static func push(note: Note, controller: UIViewController) {
+    static func push(note: Note, controller: NotesListViewController) {
         let noteEditViewController = NoteEditViewController(note: note, webView: nil)
+        noteEditViewController.listController = controller
         controller.navigationController?.pushViewController(noteEditViewController, animated: true)
     }
     

    @objc func saveHandler(sender: UIBarButtonItem) {
        self.note.text = self.textView.text
        NotesDB.shared.storeNote(note: self.note)
        self.cancelHandler(sender: sender)
    }
    

    private weak var webView: WKWebView?
    private var textView: UITextView!
    private var isModal: Bool
    private weak var listController: NotesListViewController?
    
    init(note: Note, webView: WKWebView?) {
        self.note = note
