Added smart AI

Not fully working ATM
Use OLDbattle.py for the fully working version.