Generics done

generics_2.py

@@ -8,15 +8,6 @@ from lib.gen_2 import (
 
 
 def main():
-    users = UserContainer[Employee](
-        [
-            Employee(id=1, first_name="Pero", last_name="Perić", department="Dev"),
-            Employee(id=2, first_name="Mirko", last_name="Mirković", department="Dev"),
-            Employee(id=3, first_name="Ivo", last_name="Ivić", department="Sales"),
-        ]
-    )
-    users[0]
-
     employees = EmployeeContainer(
         [
             Employee(id=1, first_name="Pero", last_name="Perić", department="Dev"),
@@ -24,7 +15,6 @@ def main():
             Employee(id=3, first_name="Ivo", last_name="Ivić", department="Sales"),
         ]
     )
-    employees[0]
 
     partners = PartnerContainer(
         [
@@ -33,12 +23,17 @@ def main():
             Partner(id=103, first_name="Mate", last_name="Matić", company="Pepsi"),
         ]
     )
-    partners[0]
 
     employee = employees.get_user_by_id(id=2)
-    print("Found user:", employee)
+    print("Found employee:", employee)
+
+    developers = employees.get_by_department("Dev")
+    print("Developers:")
+    for employee in developers:
+        print(employee)
 
     cocacola_partners = partners.get_by_company(company="CocaCola")
+    print("CocaCola partners:")
     for partner in cocacola_partners:
         print(partner)