diff --git a/docker/initialize.sh b/docker/initialize.sh
index bc10671..e1daa1a 100755
--- a/docker/initialize.sh
+++ b/docker/initialize.sh
@@ -1,7 +1,7 @@
 #!/bin/bash
 
 
-#git clone https://github.com/NVIDIA/cuda-samples
+#git clone https://github.com/NVIDIA/cuda-samples #<- To Test docker Cuda image..
 #cd cuda-samples
 
 
@@ -11,8 +11,6 @@ git checkout $(git describe --tags) # Checkout the latest release
 mkdir build
 cd build
 cmake .. -DBUILD_TESTING=OFF -DBUILD_EXAMPLES=OFF -DCMAKE_CUDA_ARCHITECTURES="60;70;80" #-DUSE_CUDA=OFF 
-
-
 make
 sudo make install
 cd ..
@@ -27,7 +25,6 @@ cd build
 cmake .. -DCMAKE_CUDA_ARCHITECTURES="60;70;80" #-DCUDA_ENABLED=OFF 
 make
 sudo make install
-#CC=/usr/bin/gcc-6 CXX=/usr/bin/g++-6 cmake ..
 cd ..
 cd ..
 
@@ -40,6 +37,7 @@ pip install  plyfile tqdm
 python3.10 -m pip install  https://huggingface.co/camenduru/gaussian-splatting/resolve/main/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
 python3.10 -m pip install  https://huggingface.co/camenduru/gaussian-splatting/resolve/main/simple_knn-0.0.0-cp310-cp310-linux_x86_64.whl
 
+python3.10 -m pip install torchvision
 
 ln -s docker/run.sh ./run.sh