Add list comprehension syntax