Update main.py

updated GUI