Fixed AI bugs

Doesn't choke when it loses, difficulty now works.
Also, sets it up for game over screen.