main.py menu bugfix