Fixed some things with the AI