more updates trying to get exit display from main.py