diff --git a/images/GCN.png b/images/GCN.png
new file mode 100644
index 0000000..915e698
Binary files /dev/null and b/images/GCN.png differ
diff --git a/main.py b/main.py
index 3d02ffc..fcf972b 100644
--- a/main.py
+++ b/main.py
@@ -24,9 +24,9 @@ __Models__ = {
 }
 
 parser = argparse.ArgumentParser(description='choose your model')
-parser.add_argument('--model', type=str, help='model name')
+parser.add_argument('--model_name', type=str, help='model name: [CNN, RNN, GCN, Capsule, Transformer, LM]')
 args = parser.parse_args()
-model_name = args.model if args.model else config.model_name
+model_name = args.model_name if args.model_name else config.model_name
 
 make_seed(config.training.seed)