1 files changed, 4 insertions, 1 deletions
diff --git a/editor.h b/editor.h
index 1ae2a51..9d2a773 100644
--- a/editor.h
+++ b/editor.h
@@ -6,20 +6,23 @@
 
 class Editor {
     public:
-        Editor(std::vector<std::string>* contents);
+        Editor(std::vector<std::string>* contents, const std::string& filename);
         int run_editor();
     private:
         const static int OUTPUT_LINE_MAX = 10;
 
         std::vector<std::string>* m_contents;
+        std::string m_filename;
         int m_current_line;
 
         int print_contents_on_line(int line) const;
         int edit_command();
         int move_to_command();
+        int write_command();
 
         int edit_line(int line);
         int move_to_line(int line);
+        int write();
 };
 
 #endif