Update TestMenu.py