From be7d40706a4d89909c2d9a5cae523c4b0d04efa4 Mon Sep 17 00:00:00 2001
From: Bruce Mayhew <mayhew64@yahoo.com>
Date: Sun, 7 Sep 2014 16:59:56 -0400
Subject: [PATCH] WEB-49 Added title to all lessons

---
 .../java/org/owasp/webgoat/lessons/AbstractLesson.java   | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/main/java/org/owasp/webgoat/lessons/AbstractLesson.java b/src/main/java/org/owasp/webgoat/lessons/AbstractLesson.java
index 24b6016b8..8b7eb338c 100644
--- a/src/main/java/org/owasp/webgoat/lessons/AbstractLesson.java
+++ b/src/main/java/org/owasp/webgoat/lessons/AbstractLesson.java
@@ -19,7 +19,9 @@ import org.apache.ecs.Element;
 import org.apache.ecs.ElementContainer;
 import org.apache.ecs.StringElement;
 import org.apache.ecs.html.Body;
+import org.apache.ecs.html.Center;
 import org.apache.ecs.html.Form;
+import org.apache.ecs.html.H1;
 import org.apache.ecs.html.Head;
 import org.apache.ecs.html.Html;
 import org.apache.ecs.html.IMG;
@@ -732,7 +734,12 @@ public abstract class AbstractLesson extends Screen implements Comparable<Object
 
         Form form = new Form(getFormAction(), Form.POST).setName("form").setEncType("");
 
-        form.addElement(createContent(s));
+		// Added this hack for WebGoat 6 to include screen titles in each lesson for Jason to screen scrape
+		ElementContainer ec = new ElementContainer();
+
+		ec.addElement(new Center(new H1(getTitle())));
+		ec.addElement(createContent(s));
+        form.addElement(ec);
 
         setContent(form);
     }