From dcc1475f36b0d58d868431a4b11b0070154d35e7 Mon Sep 17 00:00:00 2001
From: roobiselvaraj <roobiselvaraj@gmail.com>
Date: Tue, 16 Jul 2024 22:05:11 +0530
Subject: [PATCH] Update main.py

I have updated the main function
---
 main.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/main.py b/main.py
index 6261695..f268ecb 100644
--- a/main.py
+++ b/main.py
@@ -7,18 +7,24 @@ def main():
         user_input = input("Do you want to play the 3000 meter level? (yes/no) ")
         if user_input.strip().lower() == 'yes':
             my_sports.level()
+        elif user_input.strip().lower() == 'train':
+            my_sports.train()
         else:
             break
 
         user_input = input("Do you want to play the 5000 meter level? (yes/no) ")
         if user_input.strip().lower() == 'yes':
             my_sports.district()
+        elif user_input.strip().lower() == 'train':
+            my_sports.train()
         else:
             break
 
         user_input = input("Do you want to play the final match? (yes/no) ")
         if user_input.strip().lower() == 'yes':
             my_sports.state()
+        elif user_input.strip().lower() == 'train':
+            my_sports.train()
         else:
             break
 
@@ -26,6 +32,8 @@ def main():
         if user_input.strip().lower() == 'yes':
             meal_type = input("What type of meal? (light/heavy) ")
             my_sports.nutrition(meal_type)
+        elif user_input.strip().lower() == 'train':
+            my_sports.train()
 
     my_sports.event_summary()