adding more to the AI