import sys from maze import make_maze size = 8 if len(sys.argv) == 2: size = int(sys.argv[1]) m = make_maze(size) sys.stdout.write('{}'.format(m))