Working version with AI response