Update start.py

start.py

@@ -132,13 +132,19 @@ def payload_launcher(self):
             thread.start()
 
     def inject_exploit_html(self, html):
-        inject = b'<li><a href="/exploits/{EXP}/">{EXP}</a></li>\n'
-
+        inject = b'<a href="{URL}"><button class="btn btn-main">{EXP}</button></a>'
         data = b''
         try:
+            i = 0
             for exploit in os.listdir(EXPLOIT_LOC):
                 if exploit != 'index.html':
-                    data += inject.replace(b'{EXP}', bytes(exploit, 'utf-8'))
+                    if i >= 3:
+                        data += b'<br/>'
+                        i = 0
+                    url = '/exploits/{}/'.format(exploit)
+                    temp = inject.replace(b'{EXP}', bytes(exploit, 'utf-8'))
+                    data += temp.replace(b'{URL}', bytes(url, 'utf-8'))
+                    i += 1
         except IOError:
             pass