Update ants.py

Q2 Done