From 291b12d9b1bb19db11b7a69d5e1a5a99eeac1a37 Mon Sep 17 00:00:00 2001
From: Stephen McQuay <stephen@mcquay.me>
Date: Wed, 31 Oct 2012 22:33:50 -0700
Subject: [PATCH] arbirary maze sizes

---
 bin/make_maze.py | 9 +++++++--
 maze/__init__.py | 2 +-
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/bin/make_maze.py b/bin/make_maze.py
index 0ff4c98..cb7aec9 100644
--- a/bin/make_maze.py
+++ b/bin/make_maze.py
@@ -1,4 +1,9 @@
+import sys
+
 from maze import make_maze
 
-m = make_maze()
-print m
+size = 8
+if len(sys.argv) == 2:
+    size = int(sys.argv[1])
+m = make_maze(size)
+sys.stdout.write('{}'.format(m))
diff --git a/maze/__init__.py b/maze/__init__.py
index 65e421a..bf05e8c 100644
--- a/maze/__init__.py
+++ b/maze/__init__.py
@@ -44,7 +44,7 @@ class CS235Maze(object):
         return success, path
 
     def __str__(self):
-        r = derp_transpose(self._data)
+        r = derp_transpose(self._data, self._size)
         rs = ''
         for i in xrange(self._size):
             for j in xrange(self._size):