Applies AI Assistant to convert for loops into list comprehension