diff --git a/trainer/Dockerfile b/trainer/Dockerfile
index 4c06696..b707e5d 100644
--- a/trainer/Dockerfile
+++ b/trainer/Dockerfile
@@ -19,7 +19,7 @@ RUN apt update && \
     rm -rf /var/lib/apt/lists/*
 RUN python3 -m pip install --upgrade pip
 
-RUN python3 -m pip install seaborn thop coremltools onnx gsutil notebook wandb>=0.12.2 opencv-python==4.5.5.64 torchinfo python-dotenv
+RUN python3 -m pip install seaborn thop coremltools onnx gsutil notebook wandb>=0.12.2 opencv-python==4.5.5.64 torchinfo python-dotenv httptools
 
 WORKDIR /
 
diff --git a/trainer/background_detect.py b/trainer/background_detect.py
index f05729f..24f7914 100644
--- a/trainer/background_detect.py
+++ b/trainer/background_detect.py
@@ -17,6 +17,8 @@
 
 logging.basicConfig(level=logging.INFO)
 
+# NOTE: This feature is not yet ready to use and is not using the newest node lib
+
 
 @dataclass
 class Project:
diff --git a/trainer/batch_size_calculation.py b/trainer/batch_size_calculation.py
index 00b1028..27dcd3f 100644
--- a/trainer/batch_size_calculation.py
+++ b/trainer/batch_size_calculation.py
@@ -54,7 +54,7 @@ def _calc_batch_size(
     free_mem = trainer_utils.get_free_memory_mb()
     fraction = 0.95
     free_mem *= fraction
-    logging.info(f'We use only {fraction *100}% of the free memory ({free_mem})')
+    logging.info(f'{fraction:.0%} of free memory ({free_mem}) in use')
 
     device = torch.device('cuda', 0)
     torch.cuda.empty_cache()