Update ants.py

B7 done