diff --git a/docker-compose.yaml b/docker-compose.yaml
index ccd1ae7..b3f274e 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -12,7 +12,6 @@ services: # in docker compose, service are definitions of your docker containers
       target: hello-world # here you specify the build target in the Dockerfile
       x-bake: # Definitions needed for buildx bake. This enhanced docker build system makes things much easier.
         platforms:
-          #- linux/arm64 uncomment if arm64 platform is needed (like jetson nano or raspberry pi)
           - linux/amd64
 
   my_example: # for this example I am using a very simple docker build for reference: hello-world (builds very fast)
@@ -25,9 +24,18 @@ services: # in docker compose, service are definitions of your docker containers
       #target: my_example # here you specify the build target in the Dockerfile
       x-bake: # Definitions needed for buildx bake. This enhanced docker build system makes things much easier.
         platforms:
-          #- linux/arm64 uncomment if arm64 platform is needed (like jetson nano or raspberry pi)
           - linux/amd64
-          
+      environment:
+        # Allows graphical programs in the container.
+        - DISPLAY=${DISPLAY}
+        - QT_X11_NO_MITSHM=1
+        - NVIDIA_DRIVER_CAPABILITIES=all # For Nvidia hardware acceleration (CUDA, ...). This might additionally need some extra tools installed (Nvidia Container Toolkit?)
+      volumes:
+        # Allows graphical programs in the container.
+        - /tmp/.X11-unix:/tmp/.X11-unix:rw
+        - ${XAUTHORITY:-$HOME/.Xauthority}:/root/.Xauthority
+
+        
   # The rest of this file is an example of how I used docker, docker-compose and buildx bake to fully automate image building for the Segway RMP-lite 220 robot platform.
           
   # Base image containing dependencies for example the robot controller