Better ai function
Nicer graphics