Finish AI implementation