diff --git a/word_language_model/main.py b/word_language_model/main.py
index 03bfaeb87e..c60651659d 100644
--- a/word_language_model/main.py
+++ b/word_language_model/main.py
@@ -10,7 +10,7 @@
 import data
 import model
 
-parser = argparse.ArgumentParser(description='PyTorch Wikitext-2 RNN/LSTM Language Model')
+parser = argparse.ArgumentParser(description='PyTorch Wikitext-2 RNN/LSTM/GRU/Transformer Language Model')
 parser.add_argument('--data', type=str, default='./data/wikitext-2',
                     help='location of the data corpus')
 parser.add_argument('--model', type=str, default='LSTM',