Changed so the AI add properly