diff --git a/killpy/__main__.py b/killpy/__main__.py
index 58a827f..a3952cd 100644
--- a/killpy/__main__.py
+++ b/killpy/__main__.py
@@ -1,10 +1,11 @@
-import os
-from textual.app import App, ComposeResult
-from textual.widgets import DataTable, Header
-import time
 import shutil
+import time
 from pathlib import Path
 
+from textual.app import App, ComposeResult
+from textual.coordinate import Coordinate
+from textual.widgets import DataTable, Header
+
 
 def get_total_size(path: Path) -> int:
     total_size = sum(f.stat().st_size for f in path.rglob("*") if f.is_file())
@@ -35,8 +36,7 @@ def find_venvs(base_directory: Path):
 
 
 class TableApp(App):
-
-    deleted_cells = []
+    deleted_cells: Coordinate = []
 
     def compose(self) -> ComposeResult:
         yield Header()
diff --git a/pyproject.toml b/pyproject.toml
index bf0abd5..7981355 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -33,4 +33,4 @@ version_files = [
 ]
 
 [project.scripts]
-killpy = "killpy.__main__:main"
\ No newline at end of file
+killpy = "killpy.__main__:main"