diff --git a/.github/workflows/CD.yml b/.github/workflows/CD.yml
index 3b37aa4..a215965 100644
--- a/.github/workflows/CD.yml
+++ b/.github/workflows/CD.yml
@@ -56,3 +56,14 @@ jobs:
 
       - name: Clean up
         run: rm -rf build dist *.egg-info
+
+      - name: Create Release
+        id: create_release
+        uses: actions/create-release@v1
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        with:
+          tag_name: ${{ env.TAG_NAME }}
+          release_name: ${{ env.TAG_NAME }}
+          draft: false
+          prerelease: false
diff --git a/docs/RELEASES.md b/docs/RELEASES.md
index d48df1b..3ab3f92 100644
--- a/docs/RELEASES.md
+++ b/docs/RELEASES.md
@@ -1,2 +1 @@
-# 0.0.1
-First Release
\ No newline at end of file
+# 0.0.4
diff --git a/lightorch/csrc/include/lightorch/nn/cuda/deeprnn.cu b/lightorch/csrc/include/lightorch/nn/cuda/deeprnn.cu
index ae3dd0c..5053fc2 100644
--- a/lightorch/csrc/include/lightorch/nn/cuda/deeprnn.cu
+++ b/lightorch/csrc/include/lightorch/nn/cuda/deeprnn.cu
@@ -1 +1,4 @@
-// Optimized implementation for deep recurrent neural networks
\ No newline at end of file
+// Deep Neural Network
+#include <cuda.h>
+#include <torch/extension.h>
+
diff --git a/lightorch/csrc/include/lightorch/nn/deeprnn.h b/lightorch/csrc/include/lightorch/nn/deeprnn.h
index ae3dd0c..e7246df 100644
--- a/lightorch/csrc/include/lightorch/nn/deeprnn.h
+++ b/lightorch/csrc/include/lightorch/nn/deeprnn.h
@@ -1 +1 @@
-// Optimized implementation for deep recurrent neural networks
\ No newline at end of file
+// Optimized implementation for deep recurrent neural networks
diff --git a/lightorch/csrc/include/lightorch/nn/embedding.h b/lightorch/csrc/include/lightorch/nn/embedding.h
index d5cb837..7267d3b 100644
--- a/lightorch/csrc/include/lightorch/nn/embedding.h
+++ b/lightorch/csrc/include/lightorch/nn/embedding.h
@@ -3,11 +3,21 @@
 
 namespace F = torch::nn::functional;
 
-at::Tensor patch_embed_2d () {
+at::Tensor patch_embed_2d_forward () {
+    
+}
+
+at::Tensor patch_embed_2d_backward () {
+    
+}
+
+
+at::Tensor patch_embed_3d_forward () {
 
 }
 
-at::Tensor patch_embed_3d () {
+at::Tensor patch_embed_3d_backward () {
 
 }
 
+