From 9bedb55c5b31f4a3f43d89fe1d78d7a24d9003fb Mon Sep 17 00:00:00 2001
From: Emil Shanaty <65846814+emil28092005@users.noreply.github.com>
Date: Sun, 6 Apr 2025 17:54:56 +0300
Subject: [PATCH] Update get_features()

---
 main.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 5207f83..9dd6219 100644
--- a/main.py
+++ b/main.py
@@ -35,7 +35,8 @@ last_barrier = barriers[0]
 def get_features(horse):
     features = [last_barrier.rect.topleft, last_barrier.rect.bottomleft,
                last_barrier.rect.bottomright, last_barrier.rect.topright, 
-               HEIGHT, horse.rect.topleft]
+               horse.rect.topleft, horse.rect.bottomleft, horse.rect.bottomright, horse.rect.topright,
+               HEIGHT]
     return features 
 
 UI.add_horses(horses)  # Add horses to UI