Drafted more AI implementation