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):