Fixed write-to-file function.

src/optimize.py

@@ -25,3 +25,8 @@ if __name__ == '__main__':
 
     print '\nOut:'
     print out
+
+    if len(argv) > 2:
+        f = open(argv[2], 'w+')
+        f.write(out)
+        f.close()

src/writer.py

@@ -28,4 +28,6 @@ def write_to_file(filename, statements):
     '''Convert a list of statements to valid assembly code and write it to a
     file.'''
     s = write_statements(statements)
-    open(filename, 'w+').write(s).close()
+    f = open(filename, 'w+')
+    f.write(s)
+    f.close()