diff --git a/src/detector.py b/src/detector.py
index 26567be..187fc99 100644
--- a/src/detector.py
+++ b/src/detector.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python3
+"""Detect objects in images using YOLOv3."""
 
 import cv2 as cv
 import numpy as np
@@ -6,6 +6,7 @@ import sys
 import shutil
 from os import path
 import hashlib
+from dataclasses import dataclass
 
 import tempfile
 from pathlib import Path