Update CodeFileEditAtomicFlow.py

CodeFileEditAtomicFlow.py

@@ -6,6 +6,18 @@ import os
 
 
 class CodeFileEditAtomicFlow(AtomicFlow):
+    """This class is used to write code to a temp code file, with commented instructions to give information
+    to the user.
+    
+    *Input Interface*:
+    - `code`: str
+    - `language_of_code`: str
+    - `memory_files`: Dict[str, str]
+    
+    *Output Interface*:
+    - `code_editor_output`: str
+    - `temp_code_file_location`: str
+    """
     def _generate_content(self, code_lib_location, code_str) -> str:
         content = (
                 "# The below code will be appended to " +