Update main.py

ant color changes