Update menu_test.py