Update main.py

Ant colour now of owner colour