diff --git a/Dockerfile b/Dockerfile
index 7a3e0f8..0cd7c4d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,7 +1,10 @@
 # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
 FROM nvcr.io/nvidia/pytorch:20.10-py3
 
-# Install dependencies
+# Install linux packages
+RUN apt install screen
+
+# Install python dependencies
 RUN pip install --upgrade pip
 # COPY requirements.txt .
 # RUN pip install -r requirements.txt
diff --git a/utils/google_utils.py b/utils/google_utils.py
index 11d1a6c..f97349d 100644
--- a/utils/google_utils.py
+++ b/utils/google_utils.py
@@ -53,7 +53,7 @@ def attempt_download(weights):
             return
 
 
-def gdrive_download(id='1n_oKgR81BJtqk75b00eAjdv03qVCQn2f', name='coco128.zip'):
+def gdrive_download(id='1uH2BylpFxHKEGXKL6wJJlsgMU2YEjxuc', name='tmp.zip'):
     # Downloads a file from Google Drive. from utils.google_utils import *; gdrive_download()
     t = time.time()