Started to add the AI