1 files changed, 10 insertions, 3 deletions

diff --git a/main.c b/main.c
index 8fa5546..41ae0db 100644
--- a/main.c
+++ b/main.c
@@ -21,8 +21,15 @@ destroy_duplicates(Token **list, unsigned int type)
 int
 main(int argc, char **argv)
 {
-	if (argc != 2) {
-		fprintf(stderr, "usage: %s <source.golem>\n", argv[0]);
+	if (argc < 2 || argc > 3) {
+		fprintf(stderr, "usage: %s <source.golem> [gyte size]\n",
+		        argv[0]);
+		return 1;
+	}
+
+	const int gyte_size = (argc > 2) ? atoi(argv[2]) : 1;
+	if (gyte_size < 1 || gyte_size > 256) {
+		fprintf(stderr, "gyte size range is 1 to 256");
 		return 1;
 	}
 
@@ -72,7 +79,7 @@ main(int argc, char **argv)
 				fprintf(stderr, "memory_create failed\n");
 				break;
 			}
-			token_print(tokens, 1, 0);
+			//token_print(tokens, 1, 0);
 		} while(0);
 		memory_destroy(&memory);
 		token_destroy(tokens);