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()