Added better ai.