diff --git a/code/data_movement_ex.cpp b/code/data_movement_ex.cpp
index 6d45201..c523f40 100644
--- a/code/data_movement_ex.cpp
+++ b/code/data_movement_ex.cpp
@@ -35,6 +35,9 @@
         printf("%d, ", host_mem[i]);
      }
      printf("\nTask Done!\n");
+      
+     free(host_mem, my_gpu_queue);
+     free(device_mem, my_gpu_queue);  
    
      return 0;
    }