diff --git a/html/views/fileview/fileview.js b/html/views/fileview/fileview.js
index a8abad6..127e9a0 100644
--- a/html/views/fileview/fileview.js
+++ b/html/views/fileview/fileview.js
@@ -1,16 +1,20 @@
-var showFile = function(txt) {
-	$("source").style.display = "";
-	$("source").innerHTML="<pre class='first-line: 1;brush: objc'>"+txt+"</pre>";
-	
-	SyntaxHighlighter.defaults['toolbar'] = false;
-	SyntaxHighlighter.highlight();
-	
-	return;
+var showFile = function(txt, fileName) {
+  $("source").style.display = "";
+  var brush = "objc";
+  if (fileName && fileName != '') {
+    brush = fileName.substr(fileName.lastIndexOf('.') + 1);
+  }
+  $("source").innerHTML="<pre class='first-line: 1;brush: " + brush + "'>" + txt + "</pre>";
+
+  SyntaxHighlighter.defaults['toolbar'] = false;
+  SyntaxHighlighter.highlight();
+
+  return;
 }
 
 var test=function(txt) {
-	SyntaxHighlighter.defaults['toolbar'] = false;
-	SyntaxHighlighter.highlight();
-	
-	return;
+  SyntaxHighlighter.defaults['toolbar'] = false;
+  SyntaxHighlighter.highlight();
+
+  return;
 }