diff --git a/.github/workflows/unitest.yml b/.github/workflows/unitest.yml
index 57c5ebc..e693cfc 100644
--- a/.github/workflows/unitest.yml
+++ b/.github/workflows/unitest.yml
@@ -6,7 +6,7 @@ jobs:
     strategy:
       matrix:
         python-version: ["3.8", "3.9", "3.10"]
-        os: [ubuntu-20.04]
+        os: [ubuntu-latest]
       fail-fast: true
 
     steps:
@@ -17,9 +17,48 @@ jobs:
       run: |
         df -h
 
-        docker rmi `docker images -q`
-        sudo rm -rf /usr/share/dotnet /etc/mysql /etc/php /etc/apt/sources.list.d
-        sudo -E apt-get -y purge azure-cli ghc* zulu* hhvm llvm* firefox google* dotnet* powershell openjdk* mysql* php* android*
+        # Source - https://stackoverflow.com/a
+        # Posted by Cosmin Bodnariuc
+        # Retrieved 2025-11-21, License - CC BY-SA 4.0
+        sudo rm -rf /usr/share/dotnet
+        sudo rm -rf /usr/local/share/boost
+        sudo rm -rf /usr/local/share/chromium
+        sudo rm -rf /usr/local/share/powershell
+        sudo rm -rf /usr/local/share/vcpkg
+        sudo rm -rf /usr/local/share/miniconda
+        sudo rm -rf /opt/ghc
+        sudo rm -rf /opt/hostedtoolcache/CodeQL
+        sudo rm -rf /opt/hostedtoolcache/go
+        sudo rm -rf /opt/hostedtoolcache/Python
+        sudo rm -rf /opt/hostedtoolcache/node
+        sudo rm -rf /opt/hostedtoolcache/R
+        sudo rm -rf /opt/hostedtoolcache/Java
+        sudo rm -rf /opt/hostedtoolcache/LLVM
+        sudo rm -rf /opt/hostedtoolcache/Swift
+        sudo rm -rf /opt/hostedtoolcache/Php
+        sudo rm -rf /opt/hostedtoolcache/Perl
+        sudo rm -rf /opt/hostedtoolcache/Scala
+        sudo rm -rf /opt/hostedtoolcache/Julia
+        sudo rm -rf /opt/hostedtoolcache/Mono
+        sudo rm -rf /opt/hostedtoolcache/PowerShell
+        sudo rm -rf /opt/hostedtoolcache/Crystal
+        sudo rm -rf /opt/hostedtoolcache/Elixir
+        sudo rm -rf /opt/hostedtoolcache/Erlang
+        sudo rm -rf /opt/hostedtoolcache/FSharp
+        sudo rm -rf /opt/hostedtoolcache/Haskell
+        sudo rm -rf /opt/hostedtoolcache/OCaml
+        sudo rm -rf /opt/hostedtoolcache/Rust
+        sudo rm -rf /opt/hostedtoolcache/Sbt
+        sudo rm -rf /opt/hostedtoolcache/Solidity
+        sudo rm -rf /opt/hostedtoolcache/VisualStudio
+        sudo rm -rf /opt/hostedtoolcache/WinAppDriver
+        sudo rm -rf /opt/hostedtoolcache/Xamarin
+        sudo rm -rf /opt/hostedtoolcache/Yarn
+        sudo rm -rf /opt/hostedtoolcache/Zephyr
+        sudo rm -rf /opt/hostedtoolcache/zig
+        sudo rm -rf /opt/hostedtoolcache/zulu
+        sudo rm -rf /opt/hostedtoolcache/azcopy  
+
         sudo -E apt-get update
         sudo -E apt-get -y autoremove --purge
         sudo -E apt-get clean
diff --git a/requirements/dml.txt b/requirements/dml.txt
index 5e0bd35..f54a76f 100644
--- a/requirements/dml.txt
+++ b/requirements/dml.txt
@@ -43,3 +43,6 @@ python-dotenv>=1.0.0
 av
 torchfcpe
 pybase16384
+torch-directml
+torchvision
+torchaudio