Updated AI

Doesn't work yet ...