diff --git a/fleet.py b/fleet.py
index 39e52cd..06ccf0b 100644
--- a/fleet.py
+++ b/fleet.py
@@ -1,11 +1,10 @@
-def main():
-    class Fleet():
-        vehicles = {"7E19H2", "N9460G32581A3", "BN4681"}
-        def __init__(self, vtype, movspeed, travelmed, vehicleid):
-            self.vehicletype = vtype
-            self.movementspeed = movspeed
-            self.travelmedium = travelmed
-            self.uuid = vehicleid
+class Fleet():
+    vehicles = {"7E19H2", "N9460G32581A3", "BN4681"}
+    def __init__(self, vtype, movspeed, travelmed, vehicleid):
+        self.vehicletype = vtype
+        self.movementspeed = movspeed
+        self.travelmedium = travelmed
+        self.uuid = vehicleid
             
-        def displayAllFleetMembers(vehicles):
-            map(print(vehicles))
\ No newline at end of file
+    def displayAllFleetMembers(vehicles):
+        map(print(vehicles))
\ No newline at end of file