Update main.py

Revamped to include GUI