diff --git a/generics_3_without_generics.py b/generics_3_without_generics.py
index 61aa617..077b4e2 100644
--- a/generics_3_without_generics.py
+++ b/generics_3_without_generics.py
@@ -1,4 +1,4 @@
-from lib.gen_1 import (
+from lib.generics_3_without_generics import (
     Employee,
     EmployeeContainer,
     Partner,
diff --git a/generics_4_with_generics.py b/generics_4_with_generics.py
index 1e5e548..85bf672 100644
--- a/generics_4_with_generics.py
+++ b/generics_4_with_generics.py
@@ -1,9 +1,8 @@
-from lib.gen_2 import (
+from lib.generics_4_with_generics import (
     Employee,
     EmployeeContainer,
     Partner,
     PartnerContainer,
-    UserContainer,
 )
 
 
diff --git a/lib/gen_1.py b/lib/generics_3_without_generics.py
similarity index 100%
rename from lib/gen_1.py
rename to lib/generics_3_without_generics.py
diff --git a/lib/gen_2.py b/lib/generics_4_with_generics.py
similarity index 94%
rename from lib/gen_2.py
rename to lib/generics_4_with_generics.py
index d99e701..810b890 100644
--- a/lib/gen_2.py
+++ b/lib/generics_4_with_generics.py
@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from typing import Generic, List, Optional, TypeVar
+from typing import List, Optional, TypeVar
 
 
 @dataclass