diff --git a/merge_rtfs.py b/merge_rtfs.py
new file mode 100644
index 0000000..9e10af0
--- /dev/null
+++ b/merge_rtfs.py
@@ -0,0 +1,63 @@
+import glob
+import os
+
+
+# Adapted from: https://github.com/granluo/Combine-RTF
+
+def _read_file(filedir):
+    os.chdir(filedir)
+    return glob.glob("*.rtf")
+
+
+def _make_dir(ndir):
+    try:
+        os.makedirs(ndir)
+    except OSError:
+        pass
+
+
+def combine_rtf(filedir, filename, page_delimit=True):
+    """
+    filedir: directory of the file containing all rtfs needed to be combined
+    filename: the name of final combined rtf document.
+    page_delimit: determine if rtf would start in a new page or new line.
+    """
+
+    if '/' in filename:
+        _make_dir('/'.join(filename.split('/')[:-1]))
+    else:
+        _make_dir('output')
+        filename = 'output/' + filename
+
+    filenames = _read_file(filedir)
+    test = filename
+    try:
+        filenames.remove(test)
+    except ValueError:
+        pass
+    out_file = open(test, 'wb')
+    out_file.write(b"{")
+
+    for fname in filenames:
+        if test in fname:
+            continue
+
+        with open(fname, 'rb') as f1:
+            mylist = list(l1 for l1 in f1)
+            mylist[0] = mylist[0].strip()[1:]
+            mylist[-1] = mylist[-1].strip()[:-1]
+            for i in mylist:
+                out_file.write(i)
+            if page_delimit & (fname != filenames[-1]):
+                out_file.write(b"\par \page")
+
+    out_file.write(b"} ")
+    out_file.close()
+
+
+def __main():
+    combine_rtf('.', 'output.rtf', True)
+
+
+if __name__ == '__main__':
+    __main()
diff --git a/test.py b/test.py
deleted file mode 100644
index b989bdb..0000000
--- a/test.py
+++ /dev/null
@@ -1,55 +0,0 @@
-import glob, os
-
-def readfile(filedir):
-    os.chdir(filedir)
-    return glob.glob("*.rtf")
-
-def makedir(ndir):
-  try:
-    os.makedirs(ndir)
-  except OSError:
-    pass
-
-def combinertf(filedir, filename,pagedelimit = True):
-
-    """
-    filedir: directory of the file containing all rtfs needed to be combined
-    filename: the name of final combined rtf document.
-    pagedelimit: determine if rtf would start in a new page or new line.
-    """
-
-    if '/' in filename:
-        makedir('/'.join(filename.split('/')[:-1]))
-    else:
-        makedir('output')
-        filename = 'output/' + filename
-
-    filenames = readfile(filedir)
-    test = filename
-    try:
-        filenames.remove(test)
-    except ValueError:
-        pass
-    out_file = open(test,'wb')
-    out_file.write("{")
-
-    for fname in filenames:
-
-        if  test in fname:  continue
-        with open(fname, 'rb') as f1:
-            mylist = list(l1 for l1 in f1)
-            mylist[0] = mylist[0].strip()[1:]
-            mylist[-1] = mylist[-1].strip()[:-1]
-            for i in mylist:
-                out_file.write(i)
-            if pagedelimit & (fname != filenames[-1]):
-                out_file.write("\par \page")
-
-    out_file.write("} ")
-    out_file.close()
-
-def main():
-    combinertf('.','output.rtf',True)
-
-if __name__ == '__main__':
-    main()