diff --git a/src/katla.dhall b/src/katla.dhall
deleted file mode 100644
index 19d4d68..0000000
--- a/src/katla.dhall
+++ /dev/null
@@ -1,23 +0,0 @@
-{ font  = "\\ttfamily"
-, space = " "
-, datacons = { style  = ""
-             , colour = "#ed4a46"}
-, typecons = { style  = ""
-             , colour = "#368aeb"}
-, bound    = { style  = ""
-             , colour = "#dedede"}
-, function = { style  = ""
-             , colour = "#70b433"}
-, keyword  = { style  = "font-weight: bold;"
-             , colour = "#3fc5b7"}
-, comment  = { style  = ""
-             , colour = "#777777"}
-, hole     = { style  = "font-weight: bold;"
-             , colour = "#dbb32d" }
-, namespce = { style  = "font-style: italic;"
-             , colour = "#dedede"}
-, postulte = { style  = "font-weight: bold;"
-             , colour = "#ed4a46"}
-, aModule  = { style  = "font-style: italic;"
-             , colour = "#dedede"}
-}
diff --git a/src/upload.raku b/src/upload.raku
index 4a4f944..42baff4 100755
--- a/src/upload.raku
+++ b/src/upload.raku
@@ -2,6 +2,7 @@
 
 use paths;
 use File::Temp;
+use DOM::Tiny;
 use HTML::Lazy (:ALL);
 
 my $build-path;
@@ -20,34 +21,34 @@ body {
 
 # Stylize katla output
 sub stylize($katla-output) {
+    my $dom = DOM::Tiny.parse($katla-output);
+    # Insert our background css
+    $dom.find('style')[*-1][*-1].prepend($background-css);
+    # Undomify
+    my $output = $dom.Str;
     # Apply regex edits to colorize existing elements
-    $katla-output.match(/'color: darkgray'/).replace-with('color: #dedede')
-    .match(/'color: lightgrey'/).replace-with('color: #777777')
-    .match(/'color: yellow'/).replace-with('color: #3b3b3b')
-    .match(/'color: gray'/).replace-with('color: #777777')
-    .match(/'<style>'/).replace-with("<style>\n$background-css")
+    $output =
+        $output.match(/'color: darkgray'/).replace-with('color: #dedede')
+        .match(/'color: lightgrey'/).replace-with('color: #777777')
+        .match(/'color: yellow'/).replace-with('color: #3b3b3b')
+        .match(/'color: gray'/).replace-with('color: #777777');
+    $output
 }
 
 # Renders the file at the given location to the temporary directory
-sub render-idris-file($idr, $katla-config, $base-dir = "src/".IO) {
+sub render-idris-file($idr) {
     say "Converting $idr";
     # Figure out our file paths
-    my $source-file = IO::Path.new($idr, :CWD($base-dir)).resolve;
+    my $source-file = IO::Path.new($idr, :CWD("src/".IO)).resolve;
     my $ttm-file =
         IO::Path.new($idr, :CWD($build-path)).extension("ttm").resolve;
     my $out-file = IO::Path.new($idr, :CWD($dir)).extension("html").resolve;
     # Bail out now if the ttm doesn't exist
     return unless $ttm-file ~~ :e;
     # Generate the html
-    # TODO: Stop relying on the config for now and just colorize it in regexs
-    my $katla = run <katla html --config>, $katla-config,
+    my $katla = run <katla html --config>, "katla/katla.dhall",
                 $source-file, $ttm-file, :out;
     my $contents = $katla.out.slurp :close;
-    if $katla.exitcode != 0 {
-       say "Katla Error:";
-       say $contents;
-       exit $katla.exitcode;
-    }
     $contents = stylize $contents;
     # Make sure the directory exists and write the contents to it
     $out-file.dirname.IO.mkdir;
@@ -74,24 +75,13 @@ my method add-path(%index: $path, $value) {
 
 # Render the index to html
 my method render-index(%index:) {
-    # TODO sort these properly
     my @elements = [];
-    my sub compare-with-numbers($a, $b) {
-        my $a-match = $a ~~ /^(.+?)(\d*)(\.\w+)$/;
-        my $b-match = $b ~~ /^(.+?)(\d*)(\.\w+)$/;
-        ($a-match[0] cmp $b-match[0]) || ($a-match[1].Int <=> $b-match[1].Int)
-    }
-    my sub compare-keys($key1, $key2) {
-        if $key1 ~~ Str && $key2 ~~ Str {
-            compare-with-numbers($key1, $key2)
-        } elsif $key1 ~~ Str {
-            Order::Less
-        } else {
-            Order::More
-        }
-    }
-    my @keys = %index.keys.sort(&compare-keys);
-    for @keys -> $key {
+    for %index.keys -> $key {
+        say "";
+        say $key;
+        say %index{$key} ~~ Str;
+        say %index{$key};
+        say "";
         given %index{$key} {
             when Str {
                 my $x = $_;
@@ -108,12 +98,9 @@ my method render-index(%index:) {
     ul(Map, @elements)
 }
 
-#| Upload an entire project to its directory on the website
-multi MAIN(
-    $project_dir?,  #= Base directory of project to upload, defaults to $*CWD
+sub MAIN(
+    $project_dir?,  #= Base directory of project to upload, defaults to %*CWD
     $dest_dir?,     #= Destination directory on the remote
-    $katla-config = #= Configuration file to pass to katla
-    "/home/nathan/Projects/Idris/.scripts/src/katla.dhall",
 ) {
     # cd to the projet directory if needed
     if $project_dir ~~ Str {
@@ -124,30 +111,15 @@ multi MAIN(
     my @idris-files =
         paths("src", :file(-> $a {$a.ends-with(".idr") || $a.ends-with(".md")}))
             .map: *.IO.relative("src/".IO);
-    # Find and collect all of our test files
-    my @test-files =
-        paths("test/src", :file(-> $a {$a.ends-with(".idr") || $a.ends-with(".md")}))
-            .map: *.IO.relative("test/src/".IO);
-    say "Collected {@idris-files.elems} source files and {@test-files.elems} test files";
 
     # Clean if there is more than one ttc folder, easiest way to locate the
     # correct one
-    my $ttc-path = "build/ttc".IO;
-    if !($ttc-path ~~ :e) || $ttc-path.dir.elems != 1 {
-        say "Couldn't locate ttc path in $ttc-path, rebuilding";
+    if "build/ttc".IO.dir.elems > 1 {
         # Do a pack clean build so we can find the ttms
         exit -1 unless run <pack clean>;
         exit -1 unless run <pack build>;
     }
 
-    # Now do the same for the test directory
-    $ttc-path = "test/build/ttc".IO;
-    if !($ttc-path ~~ :e) || $ttc-path.dir.elems != 1 {
-        say "Couldn't locate tests ttc path in $ttc-path, rebuilding tests";
-        exit -1 unless run <pack clean test/test.ipkg>;
-        exit -1 unless run <pack build test/test.ipkg>;
-    }
-
     # find our ttc folder
     $build-path = "build/ttc/".IO.dir[0];
 
@@ -157,25 +129,12 @@ multi MAIN(
     # For each of our idris files, find the ttm, and generate the output in the
     # temporary directory
     my %index = %();
-    say "Processing idris files";
     for @idris-files.hyper :batch(1) -> $idr {
-        my $out = render-idris-file $idr, $katla-config.IO;
+        my $out = render-idris-file $idr;
         %index.&add-path($idr, $out.relative($dir)) when $out ~~ IO;
     }
 
-    # Now repeat for the test directory
-    my %test-index = %();
-    say "Processing test files";
-    $build-path = "test/build/ttc/".IO.dir[0];
-    for @test-files.hyper :batch(1) -> $idr {
-        my $out = render-idris-file $idr, $katla-config.IO, "test/src".IO;
-        %test-index.&add-path($idr, $out.relative($dir)) when $out ~~ IO;
-    }
-
-    my %final-index = ("src" => %index, "test" => ("src" => %test-index));
-
-    say "Rendering index";
-    my $index = %final-index.&render-index();
+    my $index = %index.&render-index();
     my $title = $*CWD.basename;
     $index = html-en
         head(Map,
@@ -196,13 +155,6 @@ multi MAIN(
         "Projects/$snd/$fst"
     };
 
-    say "Uploading files to $remote:$web-base/$dest";
-
     # Upload files
     run <rsync -rvh --mkpath --delete>, "$dir/", "$remote:$web-base/$dest";
 }
-
-#| Upload the highlighting of a single idris file to a content addressable store
-#| directory
-multi MAIN('file') {
-}