diff --git a/build_desktop.sh b/build_desktop.sh
index b0fa03a..7f92bb5 100755
--- a/build_desktop.sh
+++ b/build_desktop.sh
@@ -8,7 +8,7 @@ if [[ -v 1 ]]; then
     if [ "$1" = "DEBUG" ]; then
         ODIN_FLAGS="-debug"
     elif [ "$1" = "RELEASE" ]; then
-        ODIN_FLAGS="-o:minimal"
+        ODIN_FLAGS="-o:speed"
     else
         echo "Usage: <script> <DEBUG/RELEASE>"
         exit 1
@@ -19,6 +19,6 @@ else
 fi
 
 mkdir -p $OUT_DIR
-odin build src/ -out:"$OUT_DIR/$EXE_NAME" $ODIN_FLAGS
+odin build src/desktop -out:"$OUT_DIR/$EXE_NAME" $ODIN_FLAGS
 cp -R $ASSETS_DIR ./$OUT_DIR/
 echo "Desktop build created in ${OUT_DIR}"
diff --git a/build_web.sh b/build_web.sh
index 0082962..d79527c 100755
--- a/build_web.sh
+++ b/build_web.sh
@@ -4,13 +4,13 @@ ASSETS_DIR="src/assets"
 RAYLIB_PATH="src/libraries/raylib"
 EMSCRIPTEN_SDK_DIR="$HOME/programs/emsdk"
 OUT_DIR="build/web"
+INDEX_TEMPLATE_PATH="src/web/index_template.html"
 
 if [[ -v 1 ]]; then
     if [ "$1" = "DEBUG" ]; then
         ODIN_FLAGS="-debug"
     elif [ "$1" = "RELEASE" ]; then
-        # unsure if this will work
-        ODIN_FLAGS="-o:minimal"
+        ODIN_FLAGS=""
     else
         echo "Usage: <script> <DEBUG/RELEASE>"
         exit 1
@@ -32,7 +32,7 @@ export EMSDK_QUIET=1
 # The emcc call will be fed the actual raylib library file. That stuff will end
 # up in env.o
 ODIN_DEFINES="-define:RAYLIB_WASM_LIB=env.o -define:RAYGUI_WASM_LIB=env.o"
-odin build src/ -target:js_wasm32 -build-mode:obj $ODIN_DEFINES $ODIN_FLAGS -out:$OUT_DIR/$WASM_OBJ_NAME
+odin build src/web/ -target:js_wasm32 -build-mode:obj $ODIN_DEFINES $ODIN_FLAGS -out:$OUT_DIR/$WASM_OBJ_NAME
 
 ODIN_PATH=$(odin root)
 
@@ -42,10 +42,10 @@ files="$OUT_DIR/$WASM_OBJ_NAME $RAYLIB_PATH/wasm/libraylib.a $RAYLIB_PATH/wasm/l
 
 # index_template.html contains the javascript code that calls the procedures in
 # source/main_web/main_web.odin
-flags="-Oz -sASSERTIONS=0 --closure 0 -sEXPORTED_RUNTIME_METHODS=['HEAPF32'] -sUSE_GLFW=3 -sWASM_BIGINT -sWARN_ON_UNDEFINED_SYMBOLS=0 -sDECLARE_ASM_MODULE_EXPORTS=0 --shell-file index_template.html --preload-file src/assets"
+flags="-O0 -sEXPORTED_RUNTIME_METHODS=['HEAPF32'] -sUSE_GLFW=3 -sWASM_BIGINT -sWARN_ON_UNDEFINED_SYMBOLS=0 --shell-file $INDEX_TEMPLATE_PATH --preload-file src/assets"
 
 # For debugging: Add `-g` to `emcc` (gives better error callstack in chrome)
-emcc -o $OUT_DIR/index.html $files $flags
+emcc -g -o $OUT_DIR/index.html $files $flags
 
 rm -f $OUT_DIR/$WASM_OBJ_NAME
 
diff --git a/default b/default
index ac30279..f8ea14d 100755
Binary files a/default and b/default differ
diff --git a/src/animation.odin b/src/animation.odin
index f98612d..b9ae9df 100644
--- a/src/animation.odin
+++ b/src/animation.odin
@@ -1,4 +1,4 @@
-package main
+package game
 
 import rl "libraries/raylib"
 import "core:math"
diff --git a/src/desktop/main.odin b/src/desktop/main.odin
new file mode 100644
index 0000000..894e2ff
--- /dev/null
+++ b/src/desktop/main.odin
@@ -0,0 +1,19 @@
+#+build !wasm32, !wasm64p32
+
+package main
+
+import game ".."
+import "core:os"
+import "core:path/filepath"
+
+main :: proc() {
+	exe_path := os.args[0]
+	exe_dir := filepath.dir(string(exe_path))
+	os.set_working_directory(exe_dir)
+
+	game.init()
+	for game.should_run() {
+		game.update()
+	}
+	game.shutdown()
+}
diff --git a/src/grid.odin b/src/grid.odin
index 98909a9..680edee 100644
--- a/src/grid.odin
+++ b/src/grid.odin
@@ -1,4 +1,4 @@
-package main
+package game
 
 import "core:math/linalg"
 import rl "libraries/raylib"
diff --git a/src/main.odin b/src/main.odin
index 28237a7..c2748fb 100644
--- a/src/main.odin
+++ b/src/main.odin
@@ -1,4 +1,4 @@
-package main
+package game
 
 import rl "libraries/raylib"
 import "base:runtime"
diff --git a/src/main_desktop.odin b/src/main_desktop.odin
deleted file mode 100644
index 764e44c..0000000
--- a/src/main_desktop.odin
+++ /dev/null
@@ -1,18 +0,0 @@
-package main
-
-import rl "libraries/raylib"
-import "core:log"
-
-main :: proc() {
-	// exe_path := os.args[0]
-	// exe_dir := filepath.dir(string(exe_path))
-	// os.set_working_directory(exe_dir)
-	rl.ChangeDirectory(rl.GetApplicationDirectory())
-	context.logger = log.create_console_logger()
-
-	init()
-	for should_run() {
-		update()
-	}
-	shutdown()
-}
diff --git a/src/program.odin b/src/program.odin
index 1bcc499..511ba27 100644
--- a/src/program.odin
+++ b/src/program.odin
@@ -1,4 +1,4 @@
-package main
+package game
 
 /*
 	Rules of matrices:
@@ -1260,7 +1260,7 @@ ui_labeled_input_float_element :: proc(label: cstring, margin: f32, value: ^f32,
 ui_labeled_input_text_element :: proc(label: cstring, margin: f32, str: []byte, allocator := context.temp_allocator) -> ^ui.Element {
 	label := fmt.ctprintf("%s: ", label)
 	label_element := ui.element(.Label, {}, {label = label, color = ui.col(rl.WHITE), margin = margin})
-	dimensions := ui.Dimensions{ui.Size_Pixels(64.0), ui.dimensions_pixels(ui.rect_get_size(ui.get_element_outer_area(label_element^))).height}
+	dimensions := ui.Dimensions{ui.Size_Pixels(224.0), ui.dimensions_pixels(ui.rect_get_size(ui.get_element_outer_area(label_element^))).height}
 	ui.same_line()
 	input_data := new(UiElementData_TextInput, context.temp_allocator)
 	input_data^ = {str = cstring(&str[0])}
diff --git a/src/project.odin b/src/project.odin
index af8114a..8d09a98 100644
--- a/src/project.odin
+++ b/src/project.odin
@@ -1,4 +1,4 @@
-package main
+package game
 
 PROJECT_DATA_HEADER :: "WOWOWOW THIS IS HEADER WOWOWOW"
 PROJECT_DATA_VERSION :: 1
diff --git a/src/project_default.odin b/src/project_default.odin
index b4b3fa9..ec1aa19 100644
--- a/src/project_default.odin
+++ b/src/project_default.odin
@@ -1,7 +1,7 @@
 #+build !wasm32
 #+build !wasm64p32
 
-package main
+package game
 
 import "core:os"
 import "core:io"
diff --git a/src/project_web.odin b/src/project_web.odin
index 0925e86..1a35c02 100644
--- a/src/project_web.odin
+++ b/src/project_web.odin
@@ -1,5 +1,5 @@
 #+build wasm32, wasm64p32
-package main
+package game
 
 ProjectReadErrorType :: enum {
 	LoadingProjectNotSupportedOnWeb,
diff --git a/src/utils.odin b/src/utils.odin
index 1cb03af..4283955 100644
--- a/src/utils.odin
+++ b/src/utils.odin
@@ -1,4 +1,4 @@
-package main
+package game
 
 import rl "libraries/raylib"
 import "base:intrinsics"
diff --git a/src/utils_default.odin b/src/utils_default.odin
index e29681a..d30e2b9 100644
--- a/src/utils_default.odin
+++ b/src/utils_default.odin
@@ -1,7 +1,7 @@
 #+build !wasm32
 #+build !wasm64p32
 
-package main
+package game
 
 import "core:os"
 
diff --git a/src/utils_web.odin b/src/utils_web.odin
index eaabd60..52d88f5 100644
--- a/src/utils_web.odin
+++ b/src/utils_web.odin
@@ -4,7 +4,7 @@
 
 #+build wasm32, wasm64p32
 
-package main
+package game
 
 import "base:runtime"
 import "core:log"
diff --git a/index_template.html b/src/web/index_template.html
similarity index 99%
rename from index_template.html
rename to src/web/index_template.html
index ddd686c..4edec3a 100644
--- a/index_template.html
+++ b/src/web/index_template.html
@@ -45,6 +45,7 @@
 
 				return WebAssembly.instantiateStreaming(fetch("index.wasm"), newImports).then(function(output) {
 					var e = output.instance.exports;
+			        console.log(e);
 					odinMemoryInterface.setExports(e);
 					odinMemoryInterface.setMemory(e.memory);
 					return successCallback(output.instance);
diff --git a/src/main_web.odin b/src/web/main.odin
similarity index 97%
rename from src/main_web.odin
rename to src/web/main.odin
index b6b34ea..1172336 100644
--- a/src/main_web.odin
+++ b/src/web/main.odin
@@ -1,7 +1,6 @@
-#+build wasm32, wasm64p32
-
 package main
 
+import game ".."
 import "core:log"
 import "core:mem"
 import "core:c"
@@ -28,23 +27,23 @@ import "core:fmt"
 
 	web_context = context
 
-	init()
+	game.init()
 }
 
 @export main_update :: proc "c" () -> bool {
 	context = web_context
-	update()
-	return should_run()
+	game.update()
+	return game.should_run()
 }
 
 @export main_end :: proc "c" () {
 	context = web_context
-	shutdown()
+	game.shutdown()
 }
 
 @export web_window_size_changed :: proc "c" (w: c.int, h: c.int) {
 	context = web_context
-	parent_window_size_changed(int(w), int(h))
+	game.parent_window_size_changed(int(w), int(h))
 }
 
 ///////////////////////////////////////////////////////