diff --git a/flake.nix b/flake.nix
new file mode 100644
index 0000000..e17ca43
--- /dev/null
+++ b/flake.nix
@@ -0,0 +1,37 @@
+{
+  description = "MEMU";
+
+  inputs = {
+    nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
+    flake-utils.url = "github:numtide/flake-utils";
+  };
+
+  outputs = { self, nixpkgs, flake-utils }@inputs:
+    flake-utils.lib.eachDefaultSystem
+      (system:
+      let
+        pkgs = import nixpkgs { inherit system;};
+        deps = with pkgs; [
+          git
+          gnumake autoconf automake
+          cmake ninja
+          pkgsCross.riscv64-embedded.buildPackages.gcc
+        ];
+      in
+        {
+          legacyPackages = pkgs;
+          devShell = pkgs.mkShell.override { stdenv = pkgs.clangStdenv; } {
+            buildInputs = deps;
+            RV64_TOOLCHAIN_ROOT = "${pkgs.pkgsCross.riscv64-embedded.buildPackages.gcc}";
+            shellHook = ''
+							export EMU_CC=$RV64_TOOLCHAIN_ROOT/bin/riscv64-none-elf-gcc
+							export EMU_OBJCOPY=$RV64_TOOLCHAIN_ROOT/bin/riscv64-none-elf-objcopy
+							make test-img
+							unset EMU_CC
+							unset EMU_OBJCOPY
+            '';
+          };
+        }
+      )
+    // { inherit inputs;};
+}
diff --git a/src/imgfile/Makefile b/src/imgfile/Makefile
index 8df4268..47d5f4b 100644
--- a/src/imgfile/Makefile
+++ b/src/imgfile/Makefile
@@ -1,9 +1,9 @@
 CSRCS = $(wildcard *.c)
-CC ?= riscv64-linux-gnu-gcc
-OBJCOPY ?= riscv64-linux-gnu-objcopy
+EMU_CC ?= riscv64-linux-gnu-gcc
+EMU_OBJCOPY ?= riscv64-linux-gnu-objcopy
 
 compile:
 	@echo "Compiling..."
-	@riscv64-linux-gnu-gcc -march=rv64g -ffreestanding -nostdlib -static -Wl,-Ttext=0 -O2 -o prog $(CSRCS)
-	@riscv64-linux-gnu-objcopy -O binary --only-section=.text prog prog.bin
+	$(EMU_CC) -march=rv64g -ffreestanding -nostdlib -static -Wl,-Ttext=0 -O2 -o prog $(CSRCS)
+	$(EMU_OBJCOPY) -O binary --only-section=.text prog prog.bin
 	@echo "Compilation complete."