.gitignore

@@ -0,0 +1 @@
+__pycache__/

agent.py

@@ -3,10 +3,11 @@ import settings
 
 
 class Instance:
-    def __init__(self):
+    def __init__(self, tank_capacity):
         size = settings.Field.size
         self.rect = pygame.Rect(0, 0, size(), size())
         self.action = ''
+        self.tank_capacity = tank_capacity
 
     def coordinates(self):
         return {
@@ -19,7 +20,13 @@ class Instance:
 
     def y(self):
         return int(self.rect.y / settings.Field.size())
+    
+    def get_tank_capacity(self):
+        return self.tank_capacity
 
+    def set_tank_capacity(self, fuel_units):
+        self.tank_capacity = fuel_units
+        
     def move(self):
         key_pressed = pygame.key.get_pressed()
         height = settings.Pygame.height()

main.py

@@ -6,7 +6,7 @@ import settings
 import common
 import agent
 
-AGENT_IMG = pygame.image.load("dot-22-433567.png")
+AGENT_IMG = pygame.image.load('./assets/tracktor/tractor.png')
 AGENT = pygame.transform.scale(AGENT_IMG, (settings.Field.size(), settings.Field.size()))
 
 possibleFields = {
@@ -65,7 +65,7 @@ def draw_window(agent, fields):
 
 
 common = common.Instance()
-agent = agent.Instance()
+agent = agent.Instance(1000)
 
 
 def main():