Update ants.py

Implemented B5