Update main.py

Victory screen