From 5c70d23fc8f86e38078b602a71420ac60fd29175 Mon Sep 17 00:00:00 2001
From: Keerthiraj-Manickam
 <149758652+Keerthiraj-Manickam@users.noreply.github.com>
Date: Tue, 16 Jul 2024 21:03:14 +0530
Subject: [PATCH] Update main.py

---
 main.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/main.py b/main.py
index 6261695..fb3e50e 100644
--- a/main.py
+++ b/main.py
@@ -10,12 +10,22 @@ def main():
         else:
             break
 
+        rest_input = input("Do you want to rest before the next event? (yes/no) ")
+        if rest_input.strip().lower() == 'yes':
+            time_period = input("What type of rest? (short/long) ")
+            my_sports.rest(time_period)
+
         user_input = input("Do you want to play the 5000 meter level? (yes/no) ")
         if user_input.strip().lower() == 'yes':
             my_sports.district()
         else:
             break
 
+        rest_input = input("Do you want to rest before the final match? (yes/no) ")
+        if rest_input.strip().lower() == 'yes':
+            time_period = input("What type of rest? (short/long) ")
+            my_sports.rest(time_period)
+
         user_input = input("Do you want to play the final match? (yes/no) ")
         if user_input.strip().lower() == 'yes':
             my_sports.state()