Update comprehension syntax
Merge pull request #118 from arv/comprehensions

Update comprehension syntax