diff --git a/.github/workflows/rust.yml b/.github/workflows/rust.yml
index e27568c..0a025d2 100644
--- a/.github/workflows/rust.yml
+++ b/.github/workflows/rust.yml
@@ -20,6 +20,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v3
+      - uses: actions-rust-lang/setup-rust-toolchain@v1
       - uses: jetli/wasm-pack-action@v0.4.0
         with:
           version: 'latest'
@@ -29,7 +30,6 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v3
-      - uses: dtolnay/rust-toolchain@stable
       - uses: actions-rust-lang/setup-rust-toolchain@v1
         with:
           components: rustfmt
diff --git a/wasm_build.sh b/wasm_build.sh
index 2ea6702..73e148a 100755
--- a/wasm_build.sh
+++ b/wasm_build.sh
@@ -1,9 +1,13 @@
-RUSTFLAGS='-C target-feature=+atomics,+bulk-memory,+mutable-globals' wasm-pack build --target web --no-default-features --features wasm -Z build-std=panic_abort,std
+RUSTFLAGS='-C target-feature=+atomics,+bulk-memory,+mutable-globals' \
+wasm-pack build --target web --no-default-features --features \
+wasm -Z build-std=panic_abort,std && du -h pkg/proof_of_exploit_bg.wasm
 
-du -h pkg/proof_of_exploit_bg.wasm
-
-rm -rf /Users/sohamzemse/Workspace/opensource/personal/proof-of-exploit-stuff/frontend/pox-frontend/wasm
-mkdir /Users/sohamzemse/Workspace/opensource/personal/proof-of-exploit-stuff/frontend/pox-frontend/wasm
-cp -r pkg/. /Users/sohamzemse/Workspace/opensource/personal/proof-of-exploit-stuff/frontend/pox-frontend/wasm/
-rm /Users/sohamzemse/Workspace/opensource/personal/proof-of-exploit-stuff/frontend/pox-frontend/wasm/.gitignore
-rm /Users/sohamzemse/Workspace/opensource/personal/proof-of-exploit-stuff/frontend/pox-frontend/wasm/README.md
\ No newline at end of file
+if ! [ -z "$1" ]
+then
+    rm -rf $1/wasm
+    mkdir $1/wasm
+    cp -r pkg/. $1/wasm/
+    rm $1/wasm/.gitignore
+    rm $1/wasm/README.md
+    echo "Updated the package at $1/wasm"
+fi
\ No newline at end of file