I need little help with AI task