Update battle.py

Added infinite loop in idle(). Now supports multiple battles.