diff --git a/templates/index.html.ep b/templates/index.html.ep
index 08f7630..0050efe 100644
--- a/templates/index.html.ep
+++ b/templates/index.html.ep
@@ -878,10 +878,14 @@
     </div> <!-- topwrapper  -->
     <div style="clear: both;"></div>
     </div>
-    % if($training_args) {
-      <p>
-        Word vector model trained with <a href="https://code.google.com/p/word2vec/">word2vec</a> using the following parameters: <pre><%= $training_args %></pre>
-      </p>
+    % if($training_args && (@$lists)[0]) {
+      <div class="info">
+        % if($training_args =~ /-type\s*3/) {
+          Calculations are based on a word embedding model trained with an extension of <a href="https://github.com/wlin12/wang2vec/">wang2vec</a> using the following parameters: <%= $training_args %>
+        % } else {
+          Calculations are based on a word embedding model trained with <a href="https://code.google.com/p/word2vec/">word2vec</a> using the following parameters: <%= $training_args %>
+        % }
+      </div>
     % }
   </body>
 </html>