Update a console script that creates a boilerplate

Author: mdmintz

seleniumbase/console_scripts/sb_mkdir.py

@@ -152,10 +152,11 @@ def main():
         data.append("")
         data.append("    def setUp(self):")
         data.append("        super(BaseTestCase, self).setUp()")
-        data.append("        # Add custom setUp code AFTER the super() line")
+        data.append("        # << Add custom code AFTER the super() line >>")
         data.append("")
         data.append("    def tearDown(self):")
-        data.append("        # Add custom code BEFORE the super() line")
+        data.append("        self.save_teardown_screenshot()")
+        data.append("        # << Add custom code BEFORE the super() line >>")
         data.append("        super(BaseTestCase, self).tearDown()")
         data.append("")
         data.append("    def login(self):")