Add list comprehension syntax to grammar