diff --git a/Docs/Installation.md b/Docs/Installation.md
index f54f942..b103d54 100644
--- a/Docs/Installation.md
+++ b/Docs/Installation.md
@@ -16,7 +16,8 @@ interact with the containers.
 
 [Instructions to install docker GPU](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#docker)
 ````
-Do ensure that the docker command does not need sudo to run
+// Do ensure that the docker command does not need sudo to run
+sudo chmod 666 /var/run/docker.sock
 ````
 
 ### Build Project and install project
diff --git a/server/docker/docker.go b/server/docker/docker.go
index 703f7f3..061283d 100644
--- a/server/docker/docker.go
+++ b/server/docker/docker.go
@@ -257,7 +257,7 @@ func (d *DockerVM)runContainer(dockerClient *client.Client) error{
 		for i := range d.Ports {
 			cmd.WriteString("-p " + fmt.Sprint(d.Ports[i]) + ":" + fmt.Sprint(d.Ports[i]) + " ")
 		}
-		cmd.WriteString("-v=/opt/data:/data p2p-ubuntu /start > /dev/null")
+		cmd.WriteString("-v=/opt/data:/data "+ d.TagName +" /start > /dev/null")
 		//"-v=/opt/data:/data p2p-ubuntu /start > /dev/null"
 		cmdStr := cmd.String()
 		_, err := exec.Command("/bin/sh", "-c", cmdStr).Output()
diff --git a/server/docker/docker_test.go b/server/docker/docker_test.go
index 8c2a447..2c0c509 100644
--- a/server/docker/docker_test.go
+++ b/server/docker/docker_test.go
@@ -21,6 +21,15 @@ func TestDocker(t *testing.T) {
 
 }
 
+func TestContainerHorovod(t *testing.T) {
+	// Testing by providing the horovod cpu image
+	_,err := BuildRunContainer(2,"false","cpuhorovod")
+
+	if err != nil {
+		t.Error(err)
+	}
+}
+
 func TestViewAllContainers(t *testing.T) {
 	_,err := ViewAllContainers()