Adds several improvements to the AI