diff --git a/Tractor.py b/Tractor.py index 8f31ad2..4f5166f 100644 --- a/Tractor.py +++ b/Tractor.py @@ -1,6 +1,7 @@ class Tractor: - def __init__(self, horizontal_index, vertical_index, hitch, fuel_tank): + def __init__(self, horizontal_index, vertical_index, hitch, header, fuel_tank): self.horizontal_index = horizontal_index self.vertical_index = vertical_index self.hitch = hitch + self.header = header self.fuel_tank = fuel_tank diff --git a/main.py b/main.py index 40964b2..e3e6687 100644 --- a/main.py +++ b/main.py @@ -18,7 +18,7 @@ vertical_change = 0 board = Board.generate() -tractor = Tractor(horizontal_index=0, vertical_index=0, hitch="nothing", fuel_tank=100) +tractor = Tractor(horizontal_index=0, vertical_index=0, hitch="nothing", header="none", fuel_tank=100) clock = pygame.time.Clock()