Lots of improvements to all parts of the AI