diff --git a/progit/templates/docs.html b/progit/templates/docs.html
index 12556c5..ff97245 100644
--- a/progit/templates/docs.html
+++ b/progit/templates/docs.html
@@ -50,12 +50,11 @@
   </h3>
 
   {% if tree %}
-  <section class="tree_view">
+  <section class="tree_list">
       <ul>
       {% for entry in tree %}
-      <li>
+      <li class="{% if entry.filemode == 16384 %}folder{% else %}file{% endif %}">
         {% if entry.filemode == 16384 %}
-        [ ]
         <a href="{{ url_for('view_docs', username=username,
                   repo=repo.name, branchname=branchname,
                   filename=filename + '/' + entry.name if filename and '/' in filename else entry.name) }}">