Update main_menu.py